AdaptiveInvolutionNet: Spatially-adaptive involution with channel-wise attention for breast MRI tumor classification

AdaptiveInvolutionNet:一种具有通道注意力机制的空间自适应对合方法,用于乳腺MRI肿瘤分类

阅读:1

Abstract

BACKGROUND: Early and accurate classification of breast tumors from MRI scans is essential for improving patient outcomes. However, a key limitation of conventional deep learning models, such as Convolutional Neural Networks (CNNs), is their difficulty in capturing the subtle, spatially variant features that are crucial for precise medical image interpretation. OBJECTIVE: To address this limitation, we propose a novel deep learning framework called AdaptiveInvolutionNet (AIN). This hybrid architecture is specifically designed to improve discriminative feature learning for breast tumor classification by integrating two key mechanisms: spatially-adaptive involution layers and channel-wise attention. METHODS: Our AIN model employs a unique strategy for feature extraction. In its early layers, it utilizes spatially-adaptive involution kernels, which are highly effective at capturing fine-grained, localized features. As the network deepens, it transitions to conventional convolutions to maintain computational efficiency. To further enhance its diagnostic capabilities, we have embedded channel-wise attention mechanisms (specifically, squeeze-and-excitation modules) within the residual connections of the network. This allows the model to dynamically and selectively amplify diagnostically relevant features while suppressing less important ones. The model was rigorously trained and evaluated on a large, balanced dataset of 6,000 breast MRI images (3,000 benign, 3,000 malignant) using a robust five-fold cross-validation protocol. RESULTS: AIN demonstrated superior performance, achieving a high test accuracy of 97%. This performance was consistent and reliable across all folds, with an average accuracy of 96% (± 1%). The model also showed strong agreement with true labels, indicated by a high Cohen's Kappa score of 0.93 (± 0.01), and produced well-calibrated, trustworthy predictions with a low Brier score of just 0.0241. CONCLUSION: By successfully uniting an adaptive spatial feature extraction method with powerful attention mechanisms, AIN represents a significant advancement in medical image analysis. Its high accuracy, robust generalization, and consistent reliability demonstrate a strong potential for it to serve as a valuable and dependable computer-aided diagnostic tool for breast cancer detection in clinical settings.

特别声明

1、本页面内容包含部分的内容是基于公开信息的合理引用;引用内容仅为补充信息,不代表本站立场。

2、若认为本页面引用内容涉及侵权,请及时与本站联系,我们将第一时间处理。

3、其他媒体/个人如需使用本页面原创内容,需注明“来源:[生知库]”并获得授权;使用引用内容的,需自行联系原作者获得许可。

4、投稿及合作请联系:info@biocloudy.com。