Waste classification is crucial as it brings economic benefits to cities and promotes environmental sustainability. Traditional waste classification studies focus on single-label classification tasks. However, single-label classification is inefficient when dealing with images containing multiple wastes. Supervised multi-label classification can manage more complex and realistic scenes, which contain a wider variety of objects in a single image. However, it necessitates pre-labeling all waste in each image, which is time-consuming. Therefore, we design a weakly supervised multi-label classification framework called Adaptive Weakly-supervised Waste Classification Framework (AWWCF). The AWWCF consists of the Target Preprocessing Module (TPM), the Prediction Module (PM), and the Computing Module (CM). Previous weakly-supervised classification works primarily exploit inner connections among observed labels while ignoring unannotated objects and unobserved labels. To properly utilize unobserved labels, we integrate the PM and AM of AWWCF with our newly proposed method, Adaptive Loss and Enhanced Class (ALEC) activation maps. In PM, ALEC dynamically enhances the attribution scores of the class activation maps to prevent predictions for positive observed labels from being affected by unobserved labels. In the CM, ALEC dynamically rejects and corrects unobserved labels with large loss values. Experimental results demonstrate the effectiveness of our framework in weakly supervised multi-label waste classification. ALEC can be integrated with different deep classification models to form an effective framework for a sustainable environment.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Weakly Supervised Waste Classification with Adaptive Loss and Enhanced Class Activation Maps

  • Wenzhang Dai,
  • Le Sun

摘要

Waste classification is crucial as it brings economic benefits to cities and promotes environmental sustainability. Traditional waste classification studies focus on single-label classification tasks. However, single-label classification is inefficient when dealing with images containing multiple wastes. Supervised multi-label classification can manage more complex and realistic scenes, which contain a wider variety of objects in a single image. However, it necessitates pre-labeling all waste in each image, which is time-consuming. Therefore, we design a weakly supervised multi-label classification framework called Adaptive Weakly-supervised Waste Classification Framework (AWWCF). The AWWCF consists of the Target Preprocessing Module (TPM), the Prediction Module (PM), and the Computing Module (CM). Previous weakly-supervised classification works primarily exploit inner connections among observed labels while ignoring unannotated objects and unobserved labels. To properly utilize unobserved labels, we integrate the PM and AM of AWWCF with our newly proposed method, Adaptive Loss and Enhanced Class (ALEC) activation maps. In PM, ALEC dynamically enhances the attribution scores of the class activation maps to prevent predictions for positive observed labels from being affected by unobserved labels. In the CM, ALEC dynamically rejects and corrects unobserved labels with large loss values. Experimental results demonstrate the effectiveness of our framework in weakly supervised multi-label waste classification. ALEC can be integrated with different deep classification models to form an effective framework for a sustainable environment.