Development and External Validation of an Efficient Deep Learning Model for Lesion Segmentation and Subtyping of
Zhu He1, Jiajun Qiu2,3, Chenlin Du4
1School of Artificial Intelligence, Beijing University of Posts and Telecommunications Beijing China.
Background:
Hepatic echinococcosis is a zoonotic parasitic disease common in remote, resource-limited pastoral regions. It mainly appears as cystic echinococcosis (CE), due to Echinococcus granulosus, and alveolar echinococcosis (AE), caused by multilocular Echinococcus species. CE can lead to biliary colic or vascular compression, while AE often mimics liver cancer and exhibits high long-term mortality. Although ultrasonography is the diagnostic method of choice, its accuracy is affected by imaging artifacts and operator variability. Conventional deep learning models are often computationally heavy and lack interpretability, limiting their use in underdeveloped areas. Integrating frequency- and spatial-domain features in a lightweight framework offers a promising solution for efficient segmentation and classification of hepatic echinococcosis. The aim of this study is to develop efficient deep learning models for hepatic echinococcosis segmentation and classification, facilitating large-scale screening with non-invasive, portable ultrasound imaging.
Methods:
This study utilized a large ultrasound dataset to train and evaluate a deep learning model, consisting of 20,112 images from 4437 patients in Shiqu County, Sichuan Province, China, an endemic area for hepatic echinococcosis. To further assess the model's robustness, an external dataset comprising 3340 images from 1123 patients at West China Hospital of Sichuan University was used for additional testing. By enhancing the correlation of image features in both the frequency and spatial domains for hepatic echinococcosis ultrasound images, and incorporating segmentation features to assist the classification task, the developed model achieves high efficiency and lightness.
Results:
The proposed model achieved a Dice coefficient of 80.67% and 78.12% for segmentation, and classification accuracy of 90.10% and 80.96% on the internal and external test sets, respectively. Compared to the lightweight state-of-the-art (SOTA) model, it improves inference speed by 43.48% and increases classification accuracy by 8.89% and 16.32% on internal and external test sets, respectively. Compared to the standard SOTA model, it is only 8% of its size but boosts inference speed by 821.37%, with classification accuracy improvements of 3.65% and 4.57% on internal and external test sets, respectively.
Conclusions:
The proposed model offers efficient and accurate hepatic echinococcosis diagnosis, with a lightweight design suitable for both resource-limited and advanced clinical settings.
