计算机科学
分割
人工智能
计算机视觉
图像分割
人机交互
作者
Y Wang,Jin Zhang,Yihao Chen,Hongchun Yuan,Cheng Wen Wu
出处
期刊:IEEE Transactions on Industrial Informatics
[Institute of Electrical and Electronics Engineers]
日期:2024-01-01
卷期号:: 1-10
标识
DOI:10.1109/tii.2024.3353874
摘要
This article proposes an automated machine learning method for semantic segmentation that can be used for automated training of models in fields such as autonomous driving. This method is not specific to a particular semantic segmentation model. Users can simply upload a dataset with the semantic segmentation model they use, and then choose to use this method. This method implements end-to-end machine learning from sensing data to semantic segmentation results and model evaluation. It integrates four main components: unsupervised data reduction through feature extraction and clustering, a contrastive learning-based evaluator of semantic segmentation results, interactive reinforcement learning-based data selection, and automatic hyperparameter tuning through Bayesian optimization. We demonstrate the practicality of this method on the MRSI and Cityscapes datasets, and trained mainstream semantic segmentation models, such as BiSeNet and STDC. Our results show that this method can effectively guide semantic segmentation training and reduce the training time by more than 20 $\%$ .
科研通智能强力驱动
Strongly Powered by AbleSci AI