An Active Transfer Learning framework for image classification based on Maximum Differentiation Classifier

被引:0
作者
Zan, Peng [1 ]
Wang, Yuerong [1 ,2 ]
Hu, Haohao [1 ,2 ]
Zhong, Wanjun [1 ,2 ]
Han, Tianyu [1 ]
Yue, Jingwei [2 ]
机构
[1] Shanghai Univ, Sch Mechatron Engn & Automat, Shanghai Key Lab Power Stn Automat Technol, 99 Shangda Rd, Shanghai 200444, Peoples R China
[2] Beijing Inst Radiat Med, 27 Taiping Rd, Beijing 100850, Peoples R China
关键词
Active learning; Transfer learning; Image classification; Maximum differentiation classifier;
D O I
10.1016/j.imavis.2024.105401
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Deep learning has been extensively adopted across various domains, yielding satisfactory outcomes. However, it heavily relies on extensive labeled datasets, collecting data labels is expensive and time-consuming. We propose a novel framework called Active Transfer Learning (ATL) to address this issue. The ATL framework consists of Active Learning (AL) and Transfer Learning (TL). AL queries the unlabeled samples with high inconsistency by Maximum Differentiation Classifier (MDC). The MDC pulls the discrepancy between the labeled data and their augmentations to select and annotate the informative samples. Additionally, we also explore the potential of incorporating TL techniques. The TL comprises pre-training and fine-tuning. The former learns knowledge from the origin-augmentation domain to pre-train the model, while the latter leverages the acquired knowledge for the downstream tasks. The results indicate that the combination of TL and AL exhibits complementary effects, while the proposed ATL framework outperforms state-of-the-art methods in terms of accuracy, precision, recall, and F1-score.
引用
收藏
页数:10
相关论文
共 36 条
[1]   A survey on active learning and human-in-the-loop deep learning for medical image analysis [J].
Budd, Samuel ;
Robinson, Emma C. ;
Kainz, Bernhard .
MEDICAL IMAGE ANALYSIS, 2021, 71
[2]   AutoDIAL: Automatic DomaIn Alignment Layers [J].
Carlucci, Fabio Maria ;
Porzi, Lorenzo ;
Caputo, Barbara ;
Ricci, Elisa ;
Bulo, Samuel Rota .
2017 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2017, :5077-5085
[3]   Adversarial Robustness: From Self-Supervised Pre-Training to Fine-Tuning [J].
Chen, Tianlong ;
Liu, Sijia ;
Chang, Shiyu ;
Cheng, Yu ;
Amini, Lisa ;
Wang, Zhangyang .
2020 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2020, :696-705
[4]   MCDAL: Maximum Classifier Discrepancy for Active Learning [J].
Cho, Jae Won ;
Kim, Dong-Jin ;
Jung, Yunjae ;
Kweon, In So .
IEEE TRANSACTIONS ON NEURAL NETWORKS AND LEARNING SYSTEMS, 2023, 34 (11) :8753-8763
[5]   Active learning with statistical models [J].
Cohn, DA ;
Ghahramani, Z ;
Jordan, MI .
JOURNAL OF ARTIFICIAL INTELLIGENCE RESEARCH, 1996, 4 :129-145
[6]  
Deng J, 2009, PROC CVPR IEEE, P248, DOI 10.1109/CVPRW.2009.5206848
[7]  
Fu MY, 2021, AAAI CONF ARTIF INTE, V35, P7466
[8]  
Gal Y, 2017, PR MACH LEARN RES, V70
[9]  
Gal Y, 2016, PR MACH LEARN RES, V48
[10]   Rethinking ImageNet Pre-training [J].
He, Kaiming ;
Girshick, Ross ;
Dollar, Piotr .
2019 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2019), 2019, :4917-4926