Self-Supervised Bi-Channel Transformer Networks for Computer-Aided Diagnosis. 2022

Ronglin Gong, and Xiangmin Han, and Jun Wang, and Shihui Ying, and Jun Shi

Self-supervised learning (SSL) can alleviate the issue of small sample size, which has shown its effectiveness for the computer-aided diagnosis (CAD) models. However, since the conventional SSL methods share the identical backbone in both the pretext and downstream tasks, the pretext network generally cannot be well trained in the pre-training stage, if the pretext task is totally different from the downstream one. In this work, we propose a novel task-driven SSL method, namely Self-Supervised Bi-channel Transformer Networks (SSBTN), to improve the diagnostic accuracy of a CAD model by enhancing SSL flexibility. In SSBTN, we innovatively integrate two different networks for the pretext and downstream tasks, respectively, into a unified framework. Consequently, the pretext task can be flexibly designed based on the data characteristics, and the corresponding designed pretext network thus learns more effective feature representation to be transferred to the downstream network. Furthermore, a transformer-based transfer module is developed to efficiently enhance knowledge transfer by conducting feature alignment between two different networks. The proposed SSBTN is evaluated on two publicly available datasets, namely the full-field digital mammography INbreast dataset and the wireless video capsule CrohnIPI dataset. The experimental results indicate that the proposed SSBTN outperforms all the compared algorithms.

UI MeSH Term Description Entries
D008327 Mammography Radiographic examination of the breast. 3D-Mammography,Digital Breast Tomosynthesis,Digital Mammography,X-ray Breast Tomosynthesis,3D Mammography,3D-Mammographies,Breast Tomosyntheses, Digital,Breast Tomosyntheses, X-ray,Breast Tomosynthesis, Digital,Breast Tomosynthesis, X-ray,Digital Breast Tomosyntheses,Digital Mammographies,Mammographies,Mammographies, Digital,Mammography, Digital,X ray Breast Tomosynthesis,X-ray Breast Tomosyntheses
D003201 Computers Programmable electronic devices designed to accept data, perform prescribed mathematical and logical operations at high speed, and display the results of these operations. Calculators, Programmable,Computer Hardware,Computers, Digital,Hardware, Computer,Calculator, Programmable,Computer,Computer, Digital,Digital Computer,Digital Computers,Programmable Calculator,Programmable Calculators
D003936 Diagnosis, Computer-Assisted Application of computer programs designed to assist the physician in solving a diagnostic problem. Computer-Assisted Diagnosis,Computer Assisted Diagnosis,Computer-Assisted Diagnoses,Diagnoses, Computer-Assisted,Diagnosis, Computer Assisted
D006801 Humans Members of the species Homo sapiens. Homo sapiens,Man (Taxonomy),Human,Man, Modern,Modern Man
D000465 Algorithms A procedure consisting of a sequence of algebraic formulas and/or logical steps to calculate or determine a given task. Algorithm
D016571 Neural Networks, Computer A computer architecture, implementable in either hardware or software, modeled after biological neural networks. Like the biological system in which the processing capability is a result of the interconnection strengths between arrays of nonlinear processing nodes, computerized neural networks, often called perceptrons or multilayer connectionist models, consist of neuron-like units. A homogeneous group of units makes up a layer. These networks are good at pattern recognition. They are adaptive, performing tasks by example, and thus are better for decision-making than are linear learning machines or cluster analysis. They do not require explicit programming. Computational Neural Networks,Connectionist Models,Models, Neural Network,Neural Network Models,Neural Networks (Computer),Perceptrons,Computational Neural Network,Computer Neural Network,Computer Neural Networks,Connectionist Model,Model, Connectionist,Model, Neural Network,Models, Connectionist,Network Model, Neural,Network Models, Neural,Network, Computational Neural,Network, Computer Neural,Network, Neural (Computer),Networks, Computational Neural,Networks, Computer Neural,Networks, Neural (Computer),Neural Network (Computer),Neural Network Model,Neural Network, Computational,Neural Network, Computer,Neural Networks, Computational,Perceptron

Related Publications

Ronglin Gong, and Xiangmin Han, and Jun Wang, and Shihui Ying, and Jun Shi
October 2023, Computers in biology and medicine,
Ronglin Gong, and Xiangmin Han, and Jun Wang, and Shihui Ying, and Jun Shi
December 2023, IEEE journal of biomedical and health informatics,
Ronglin Gong, and Xiangmin Han, and Jun Wang, and Shihui Ying, and Jun Shi
August 2022, IEEE transactions on medical imaging,
Ronglin Gong, and Xiangmin Han, and Jun Wang, and Shihui Ying, and Jun Shi
January 2023, Frontiers in artificial intelligence,
Ronglin Gong, and Xiangmin Han, and Jun Wang, and Shihui Ying, and Jun Shi
August 2007, Academic radiology,
Ronglin Gong, and Xiangmin Han, and Jun Wang, and Shihui Ying, and Jun Shi
July 2019, Annual International Conference of the IEEE Engineering in Medicine and Biology Society. IEEE Engineering in Medicine and Biology Society. Annual International Conference,
Ronglin Gong, and Xiangmin Han, and Jun Wang, and Shihui Ying, and Jun Shi
December 2023, IEEE transactions on pattern analysis and machine intelligence,
Ronglin Gong, and Xiangmin Han, and Jun Wang, and Shihui Ying, and Jun Shi
June 2020, Physics in medicine and biology,
Ronglin Gong, and Xiangmin Han, and Jun Wang, and Shihui Ying, and Jun Shi
January 2022, Computational and mathematical methods in medicine,
Ronglin Gong, and Xiangmin Han, and Jun Wang, and Shihui Ying, and Jun Shi
July 2013, Computer methods and programs in biomedicine,
Copied contents to your clipboard!