More Reliable AI Solution: Breast Ultrasound Diagnosis Using Multi-AI Combination

Jian Dai,Shuge Lei,Licong Dong,Xiaona Lin,Huabin Zhang,Desheng Sun,Kehong Yuan
DOI: https://doi.org/10.48550/arXiv.2101.02639
2021-07-12
Abstract:Objective: Breast cancer screening is of great significance in contemporary women's health prevention. The existing machines embedded in the AI system do not reach the accuracy that clinicians hope. How to make intelligent systems more reliable is a common problem. Methods: 1) Ultrasound image super-resolution: the SRGAN super-resolution network reduces the unclearness of ultrasound images caused by the device itself and improves the accuracy and generalization of the detection model. 2) In response to the needs of medical images, we have improved the YOLOv4 and the CenterNet models. 3) Multi-AI model: based on the respective advantages of different AI models, we employ two AI models to determine clinical resuls cross validation. And we accept the same results and refuses others. Results: 1) With the help of the super-resolution model, the YOLOv4 model and the CenterNet model both increased the mAP score by 9.6% and 13.8%. 2) Two methods for transforming the target model into a classification model are proposed. And the unified output is in a specified format to facilitate the call of the molti-AI model. 3) In the classification evaluation experiment, concatenated by the YOLOv4 model (sensitivity 57.73%, specificity 90.08%) and the CenterNet model (sensitivity 62.64%, specificity 92.54%), the multi-AI model will refuse to make judgments on 23.55% of the input data. Correspondingly, the performance has been greatly improved to 95.91% for the sensitivity and 96.02% for the specificity. Conclusion: Our work makes the AI model more reliable in medical image diagnosis. Significance: 1) The proposed method makes the target detection model more suitable for diagnosing breast ultrasound images. 2) It provides a new idea for artificial intelligence in medical diagnosis, which can more conveniently introduce target detection models from other fields to serve medical lesion screening.
Computer Vision and Pattern Recognition
What problem does this paper attempt to address?
### What problems does this paper attempt to solve? This paper aims to solve the practicality and reliability problems of artificial intelligence (AI) systems in breast cancer screening. Specifically, most of the current AI systems used for breast cancer screening fail to meet clinical standards, especially when dealing with ultrasound images of different resolutions. In addition, existing AI systems are usually based on a single traditional detection model, which limits their performance and generalization ability in practical applications. To solve these problems, the author proposes a more practical ultrasound computer - aided diagnosis system - the Multi - AI Combination detection system. This system improves the accuracy and reliability of breast ultrasound diagnosis through the following methods: 1. **Super - Resolution Generative Adversarial Network (SRGAN)**: It is used to improve the clarity of ultrasound images, reduce image blurring caused by the device itself, and thus improve the accuracy and generalization ability of the detection model. 2. **Improved object detection models**: The YOLOv4 and CenterNet models are improved to better adapt to medical standards. 3. **Multi - AI model integration system**: Based on the advantages of different AI models, the clinical results are detected by cross - validation, and only the results consistent with the two models are accepted. This method not only improves the accuracy of classification tasks but also ensures the reliability of the system. ### Main challenges 1. **Insufficient computing power**: The computing power of existing AI - embedded devices is limited, and it is difficult to achieve real - time computing, especially under the requirements of high - precision medical diagnosis. 2. **Resolution differences**: Current systems can usually only process images within a specific resolution range, resulting in limited generalization ability and scalability of the system. 3. **Limitations of a single model**: Most systems rely on a single traditional detection model, which performs poorly on individual evaluation indicators. ### Solutions - **Super - resolution processing**: Use the SRGAN network to improve image clarity, making subsequent analysis easier. - **Multi - model integration**: Combine the advantages of multiple AI models and improve the accuracy and reliability of diagnosis through cross - validation. - **Cloud computing support**: Utilize the high computing power of the cloud platform to help inexperienced operators improve their diagnostic accuracy and allow experienced doctors to focus on complex images. ### Experimental results The experimental results show that by introducing the super - resolution model, the mAP scores of the YOLOv4 and CenterNet models are increased by 9.6% and 13.8% respectively. The multi - AI model rejects 23.55% of the input data in the classification evaluation experiment, but significantly improves the sensitivity and specificity, reaching 95.91% and 96.02% respectively. In conclusion, the multi - AI combination detection system proposed in this study shows higher practicality and reliability in breast ultrasound diagnosis and has important clinical significance.