Isak Schwartz and William Åkvist defended their Master thesis in MPALG and MPCAS at the Chalmers University of Technology on 1 June 2022 at 16:00. Congrats!
Title: Active learning in deep convolutional neural networks for image segmentation
Evaluating data-centric approaches to improving performance in seat belt localization from images
The sitting position and seat belt orientation of passengers in automobiles can be crucial in the event of a collision. In order to warn passengers of unsafe positions, deep learning models in the form of neural networks can be used to identify the seat belt from image data. Performance of neural networks can be increased by improving the model (model-centric approaches) or by improving the data used to train the model (data-centric approaches).
In this thesis we compare the segmentation performance gains from model-centric approaches to data-centric approaches including stratified sampling, balancing, label error reduction and active learning. Active learning is the process of iteratively choosing data points for labeling according to the expected improvement in model performance. No new model architecture was found that improved performance, but the model training time was sped up by four times without performance loss. Stratified sampling, balancing and error reduction did not improve performance.
In active learning, images to be labeled were selected according to the model’s uncertainty. Several uncertainty metrics were used, all leading to an improvement when using active learning. The best result showed that we achieved 95% and 99% of the best baseline performance using 19% and 23% less data respectively.
Name of the master programme: MPALG – Computer Science: Algorithms, Languages, and Logic, MPCAS – Complex Adaptive Systems
Examiner: Giovanni Volpe
Supervisor: Tomas Björklund, Sheng Huang (Volvo Cars Corporation)
Opponent: Rohini Bisht, Selomie Kindu
Time: 1 June, 2022, 16:00