An efficient computational framework for gastrointestinal disorder prediction using attention-based transfer learning

被引:0
|
作者
Zhou, Jiajie [1 ]
Song, Wei [1 ]
Liu, Yeliu [1 ]
Yuan, Xiaoming [1 ]
机构
[1] Nanjing Med Univ, Huaian Peoples Hosp 1, Nanjing, Jiangsu, Peoples R China
关键词
Gastrointestinal disorders; Deep learning; Attention; Transfer learning; Data science; EOSINOPHILIC ESOPHAGITIS; ARTIFICIAL-INTELLIGENCE; CANCER STATISTICS; NEURAL-NETWORK; ENDOSCOPY; DIAGNOSIS; CLASSIFICATION;
D O I
10.7717/peerj-cs.2059
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Diagnosing gastrointestinal (GI) disorders, which affect parts of the digestive system such as the stomach and intestines, can be difficult even for experienced gastroenterologists due to the variety of ways these conditions present. Early diagnosis is critical for successful treatment, but the review process is time-consuming and labor-intensive. Computer -aided diagnostic (CAD) methods provide a solution by automating diagnosis, saving time, reducing workload, and lowering the likelihood of missing critical signs. In recent years, machine learning and deep learning approaches have been used to develop many CAD systems to address this issue. However, existing systems need to be improved for better safety and reliability on larger datasets before they can be used in medical diagnostics. In our study, we developed an effective CAD system for classifying eight types of GI images by combining transfer learning with an attention mechanism. Our experimental results show that ConvNeXt is an effective pre -trained network for feature extraction, and ConvNeXt+Attention (our proposed method) is a robust CAD system that outperforms other cutting -edge approaches. Our proposed method had an area under the receiver operating characteristic curve of 0.9997 and an area under the precision -recall curve of 0.9973, indicating excellent performance. The conclusion regarding the effectiveness of the system was also supported by the values of other evaluation metrics.
引用
收藏
页数:19
相关论文
共 50 条
  • [1] An integrated federated learning with CRSO of attention-based LSTM framework for efficient IoT DataStream prediction
    El-Saied, Asma M.
    [J]. MULTISCALE AND MULTIDISCIPLINARY MODELING EXPERIMENTS AND DESIGN, 2024,
  • [2] An Attention-Based Deep Learning Framework for Trip Destination Prediction of Sharing Bike
    Wang, Wei
    Zhao, Xiaofeng
    Gong, Zhiguo
    Chen, Zhikui
    Zhang, Ning
    Wei, Wei
    [J]. IEEE TRANSACTIONS ON INTELLIGENT TRANSPORTATION SYSTEMS, 2021, 22 (07) : 4601 - 4610
  • [3] Attention-Based Deep Learning Framework for Hemiplegic Gait Prediction With Smartphone Sensors
    Thakur, Dipanwita
    Biswas, Suparna
    [J]. IEEE SENSORS JOURNAL, 2022, 22 (12) : 11979 - 11988
  • [4] Attention-based learning
    Kasderidis, S
    Taylor, JG
    [J]. 2004 IEEE INTERNATIONAL JOINT CONFERENCE ON NEURAL NETWORKS, VOLS 1-4, PROCEEDINGS, 2004, : 525 - 530
  • [5] Spatio-Temporal Attention-Based Deep Learning Framework for Mesoscale Eddy Trajectory Prediction
    Wang, Xuegong
    Li, Chong
    Wang, Xinning
    Tan, Lining
    Wu, Jin
    [J]. IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2022, 15 : 3853 - 3867
  • [6] A novel attention-based cross-modal transfer learning framework for predicting cardiovascular disease
    Prakash, V. Jothi
    Vijay, S. Arul Antran
    Kumar, P. Ganesh
    Karthikeyan, N. K.
    [J]. COMPUTERS IN BIOLOGY AND MEDICINE, 2024, 170
  • [7] Attention-Based Transfer Learning for Efficient Pneumonia Detection in Chest X-ray Images
    Cha, So-Mi
    Lee, Seung-Seok
    Ko, Bonggyun
    [J]. APPLIED SCIENCES-BASEL, 2021, 11 (03): : 1 - 15
  • [8] Dual Attention-Based Federated Learning for Wireless Traffic Prediction
    Zhang, Chuanting
    Dang, Shuping
    Shihada, Basem
    Alouini, Mohamed-Slim
    [J]. IEEE CONFERENCE ON COMPUTER COMMUNICATIONS (IEEE INFOCOM 2021), 2021,
  • [9] Effective ensemble learning approach for SST field prediction using attention-based PredRNN
    Qiao, Baiyou
    Wu, Zhongqiang
    Ma, Ling
    Zhou, Yicheng
    Sun, Yunjiao
    [J]. FRONTIERS OF COMPUTER SCIENCE, 2023, 17 (01)
  • [10] Learning Attention-based Embeddings for Relation Prediction in Knowledge Graphs
    Nathani, Deepak
    Chauhan, Jatin
    Sharma, Charu
    Kaul, Manohar
    [J]. 57TH ANNUAL MEETING OF THE ASSOCIATION FOR COMPUTATIONAL LINGUISTICS (ACL 2019), 2019, : 4710 - 4723