A Multi-objective Optimization Benchmark Test Suite for Real-time Semantic Segmentation

Yifan Zhao,Zhenyu Liang,Zhichao Lu,Ran Cheng
DOI: https://doi.org/10.1145/3638530.3654389
2024-04-29
Abstract:As one of the emerging challenges in Automated Machine Learning, the Hardware-aware Neural Architecture Search (HW-NAS) tasks can be treated as black-box multi-objective optimization problems (MOPs). An important application of HW-NAS is real-time semantic segmentation, which plays a pivotal role in autonomous driving scenarios. The HW-NAS for real-time semantic segmentation inherently needs to balance multiple optimization objectives, including model accuracy, inference speed, and hardware-specific considerations. Despite its importance, benchmarks have yet to be developed to frame such a challenging task as multi-objective optimization. To bridge the gap, we introduce a tailored streamline to transform the task of HW-NAS for real-time semantic segmentation into standard MOPs. Building upon the streamline, we present a benchmark test suite, CitySeg/MOP, comprising fifteen MOPs derived from the Cityscapes dataset. The CitySeg/MOP test suite is integrated into the EvoXBench platform to provide seamless interfaces with various programming languages (e.g., Python and MATLAB) for instant fitness evaluations. We comprehensively assessed the CitySeg/MOP test suite on various multi-objective evolutionary algorithms, showcasing its versatility and practicality. Source codes are available at
Computer Vision and Pattern Recognition,Neural and Evolutionary Computing
What problem does this paper attempt to address?
The main goal of this paper is to address the multi-objective optimization problem of hardware-aware neural architecture search (HW-NAS) in real-time semantic segmentation. Specifically, the research team developed a multi-objective optimization benchmark suite named CitySeg/MOP, aimed at optimizing real-time semantic segmentation tasks. In real-time applications (such as autonomous driving), it is necessary to ensure model accuracy while also considering computational efficiency and hardware-specific factors, such as inference speed and energy consumption. These requirements make real-time semantic segmentation a complex multi-objective optimization problem. Although hardware-aware neural architecture search (HW-NAS) has become increasingly important in this field, there is currently a lack of a suitable multi-objective optimization benchmark to evaluate the effectiveness of different methods. To address this issue, the authors propose a method to transform the HW-NAS problem into a standard multi-objective optimization problem and based on this, constructed the CitySeg/MOP benchmark suite. This suite includes 15 different multi-objective optimization problem instances derived from the Cityscapes dataset, covering different numbers of objectives and hardware platforms. Additionally, CitySeg/MOP is integrated into the EvoXBench platform to support multiple programming languages (such as Python and MATLAB), enabling rapid performance evaluation. Through experimental evaluation, CitySeg/MOP demonstrated its effectiveness in terms of prediction accuracy, sample diversity, and evaluation efficiency. It was also used to evaluate the performance of six representative multi-objective evolutionary algorithms, showing that different algorithms have different advantages when dealing with problems with varying numbers of objectives. Overall, CitySeg/MOP provides a comprehensive and practical benchmark framework for the multi-objective optimization problem in real-time semantic segmentation.