TSLA: A Task-Specific Learning Adaptation for Semantic Segmentation on Autonomous Vehicles Platform
Autonomous driving platforms encounter diverse driving scenarios, each with varying hardware resources and precision requirements. Given the computational limitations of embedded devices, it is crucial to consider computing costs when deploying on target platforms like the DRIVE PX 2. Our objective is to customize the semantic segmentation network according to the computing power and specific scenarios of autonomous driving hardware. We implement dynamic adaptability through a three-tier control mechanism—width multiplier, classifier depth, and classifier kernel—allowing fine-grained control over model components based on hardware constraints and task requirements. This adaptability facilitates broad model scaling, targeted refinement of the final layers, and scenario-specific optimization of kernel sizes, leading to improved resource allocation and performance. Additionally, we leverage Bayesian Optimization with surrogate modeling to efficiently explore hyperparameter spaces under tight computational budgets. Our approach addresses scenario-specific and task-specific requirements through automatic parameter search, accommodating the unique computational complexity and accuracy needs of autonomous driving. It scales its multiply-accumulate operations (MACs) for task-specific learning adaptation (TSLA), resulting in alternative configurations tailored to diverse self-driving tasks. These TSLA customizations maximize computational capacity and model accuracy, optimizing hardware utilization.
Authors
- Geng Yuan (ORCID: https://orcid.org/0000-0001-9844-992X)
- Weihao Zeng (ORCID: https://orcid.org/0009-0005-9314-2415)
- Changdi Yang (ORCID: https://orcid.org/0000-0002-8848-3806)
- Wei Niu (ORCID: https://orcid.org/0000-0002-2697-7042)
- Xue Lin (ORCID: https://orcid.org/0000-0001-6210-8883)
- Hao Tang (ORCID: https://orcid.org/0000-0002-2077-1246)
- Pu Zhao (ORCID: https://orcid.org/0000-0001-5018-2859)
- Zhenglun Kong (ORCID: https://orcid.org/0000-0002-8120-4456)
- Jun Liu (ORCID: https://orcid.org/0000-0003-3808-4599)
- Yanzhi Wang (ORCID: https://orcid.org/0000-0002-3024-7990)
- Xuan Shen (ORCID: https://orcid.org/0000-0003-4965-7321)
- Wenbin Zhang
Institutions
- Northeastern University (US)
- University of Georgia (US)
- Florida International University (US)
- Carnegie Mellon University (US)
Publication Details
- Journal
- IEEE Transactions on Computer-Aided Design of Integrated Circuits and Systems
- Published
- 2024-11-04
- DOI
- https://doi.org/10.1109/tcad.2024.3491015
- Citations
- 3
- Primary Topic
- Advanced Neural Network Applications
- Type
- article
- Field-Weighted Citation Impact
- 0.75