1. Introduction
Breast cancer remains one of the most common malignancies among women and is a major cause of cancer-related mortality worldwide. Early and accurate detection of breast lesions is therefore essential for improving treatment planning and patient outcomes. Mammography is widely used for breast cancer screening and diagnosis; however, interpretation remains challenging because lesion appearance varies with breast density, lesion morphology, contrast, and imaging quality.
Manual interpretation of mammograms is time-consuming and may be affected by inter-observer variability, especially when lesions are small, low-contrast, irregular, or obscured by dense breast tissue. These challenges have motivated the development of computer-aided diagnosis systems based on machine learning and deep learning [
1,
2]. Although convolutional neural networks have shown strong performance in medical image classification and segmentation, their local receptive fields may limit their ability to model long-range contextual relationships [
3]. Transformer-based architectures, particularly Swin Transformers, provide an effective mechanism for hierarchical contextual representation, but they may require additional architectural support to preserve local boundary information [
4,
5]. Existing Swin Transformer-based segmentation frameworks commonly employ a U-shaped architecture in which hierarchical encoder representations are transferred to a comparatively large decoder through direct skip connections. Hybrid CNN–Transformer models usually employ a separate convolutional pathway to recover local texture and boundary information and subsequently combine it with transformer-derived contextual features through addition or concatenation. Although these approaches are effective, direct fusion may assign equal importance to feature levels with different semantic and spatial characteristics, while parallel encoders and full segmentation decoders increase architectural complexity.
To address these limitations, this study proposes MSF-Swin-ICFNet, an image-derived Swin Transformer framework with an explicit Image-Derived Cross-Scale Fusion (ICF) mechanism. The proposed fusion strategy differs from conventional Swin Transformer encoder–decoder architectures in which hierarchical encoder features are transferred directly to corresponding decoder stages through skip connections. It also differs from hybrid CNN–Transformer approaches that extract local and contextual representations through separate branches and subsequently combine them using fixed addition or concatenation [
6]. In MSF-Swin-ICFNet, feature maps from all four Swin Transformer stages are first projected and spatially aligned to a common feature space, their relative contributions are adaptively recalibrated, and the resulting scale-weighted representations are fused into a shared image-derived feature representation [
7,
8,
9,
10,
11]. Thus, the proposed strategy uses the complete hierarchical Swin feature pyramid without requiring a parallel CNN encoder or full symmetric segmentation decoder. The fused representation is subsequently shared by the benign–malignant classification head and the lightweight lesion-localization head.
Recent breast imaging studies have increasingly adopted transformer-based and hybrid CNN–Transformer models to combine local texture representation with global contextual modelling [
12,
13,
14,
15,
16,
17,
18,
19,
20,
21,
22]. However, mammographic lesion analysis remains difficult because small masses, calcifications, irregular margins, and dense tissue patterns may appear at different spatial scales. Therefore, a model designed for mammography should preserve fine boundary cues while also learning broader contextual relationships across the image. The proposed MSF-Swin-ICFNet framework addresses this requirement through hierarchical Swin Transformer feature extraction, image-derived cross-scale fusion, lightweight feature refinement, and a boundary-aware mask head.
In the proposed framework, segmentation is used as a complementary lesion-localization output rather than as an independent diagnostic endpoint. The predicted mask provides spatial information about the image region associated with the model’s lesion representation and can therefore be overlaid on the original mammogram to indicate the approximate location and spatial extent of the suspected lesion. From an interpretability perspective, this output allows a radiologist to examine whether a benign–malignant classification is supported by a spatially plausible abnormal region rather than relying solely on the predicted probability. The localization map may also help identify discordant cases, for example when a high malignancy probability is produced without a clearly localized lesion or when the predicted region extends substantially beyond the visually suspicious area. In this way, the segmentation output may provide supplementary image-based evidence during a retrospective review and may support interpretation of the model’s classification decision. However, the generated masks are not intended to replace radiologist lesion delineation, BI-RADS assessment, or pathology-based diagnosis, and their value for clinical decision-making must be established through independent external validation and radiologist reader studies.
The primary contributions of this research are as follows:
1.1. Contributions
An Image-Derived Cross-Scale Fusion mechanism that aligns and adaptively recalibrates four hierarchical Swin Transformer feature levels.
A shared fused representation for benign–malignant classification and lightweight lesion localization without a parallel CNN encoder or full symmetric decoder.
Component-wise evaluation of cross-scale fusion, scale recalibration, feature refinement and boundary-aware mask prediction
Evaluation of a public benchmark and a clinically provided de-identified image collection, with the evaluation units and dataset limitations explicitly reported.
Unlike SwinCAMF-Net, which performs multimodal fusion of mammographic, volumetric, and clinical information through cross-attention, the proposed MSF-Swin-ICFNet focuses on image-based hierarchical feature integration across multiple Swin stages. The proposed framework is therefore designed to improve cross-scale feature representation without requiring additional volumetric or clinical modalities.
A direct architectural comparison is made with the previously published SwinCAMF-Net framework to establish that the present method addresses a distinct image-derived feature-fusion setting.
1.2. Novelty Aspects of the Proposed MSF-Swin-ICFNet Model
The scientific contribution of MSF-Swin-ICFNet does not arise merely from excluding clinical metadata or volumetric ROI inputs. Its principal technical contribution is the Image-Derived Cross-Scale Fusion (ICF) mechanism, which operates directly on the complete four-stage hierarchical feature pyramid generated by the Swin Transformer. Unlike approaches that use only the deepest representation, direct skip transfer, or fixed addition/concatenation of selected feature levels, ICF first transforms the four heterogeneous Swin representations into a common feature space, spatially aligns them, estimates the relative contribution of each scale, and performs adaptive scale-weighted fusion. Consequently, shallow representations containing comparatively fine lesion-boundary information and deeper representations containing higher-level semantic context contribute to the fused representation according to learned scale importance rather than being combined with equal predetermined weights.
A second distinguishing aspect is that the resulting fused representation is shared by two complementary tasks: benign–malignant classification and lightweight boundary-aware lesion localization. Thus, the framework does not require a parallel CNN feature extractor or a full symmetric encoder–decoder segmentation pathway. The technical novelty therefore lies in the combination of (i) complete four-stage hierarchical Swin feature utilization; (ii) explicit cross-scale alignment; (iii) adaptive scale recalibration prior to fusion; and (iv) shared use of the recalibrated fused representation for classification and lightweight localization. Removal of multimodal inputs should consequently be viewed as a difference in the intended input setting rather than as the primary source of novelty.
1.3. Relationship with Previous Work
The authors previously published a related article entitled “SwinCAMF-Net: Explainable Cross-Attention Multimodal Swin Network for Mammogram Analysis” in Diagnostics in 2025. That study proposed a multimodal framework integrating mammographic views, ROI-derived volume representations, and clinical metadata using a cross-attention fusion module. The present study is related to clinical problem setting and dataset usage, but it is architecturally different in scope and design. The present framework is an image-derived cross-scale fusion model that does not use clinical metadata, clinical projection, or volumetric ROI input. Its focus is to investigate whether multi-resolution mammographic image features alone can support breast lesion classification and segmentation through cross-scale feature integration and a lightweight mask head. Therefore, the present manuscript should be interpreted as a related but distinct image-derived extension, not as a duplicate of the earlier multimodal SwinCAMF-Net work.
Table 1 presents the difference between the SwinCAMF-Net study and the present framework in the manuscript.
2. Literature Review
Breast cancer remains a major public health concern among women, and mammography is one of the most widely used imaging modalities for screening and early diagnosis. Despite its clinical value, mammogram interpretation is challenging because benign and malignant lesions may show overlapping features in shape, density, margin appearance, and contrast. Dense breast tissue, anatomical overlap, and subtle lesion boundaries further complicate automated detection and classification. As a result, to help radiologists find breast lesions more easily to classify and segment those lesions, computer-aided diagnosis (CAD) systems have been researched and developed extensively.
Publicly available mammography datasets have contributed significantly to the advancement of computer-aided diagnosis of breast cancer. The CBIS-DDSM is one of the most widely used benchmark datasets for the research of automated breast cancer diagnosis. CBIS-DDSM is a curated digital version of the Digital Database for Screening Mammography (DDSM) and contains mammograms of breast lesions with annotations and pathology-related data. While DDSM contains normal, benign, and malignant digital mammographic examinations, the CBIS-DDSM includes improved data curation, updated lesion segmentations, bounding boxes, and pathology codes, making the CBIS-DDSM data ideal for the development and evaluation of deep learning-based methods for the diagnosis of breast cancer.
Early CAD systems relied on handcrafted descriptors such as shape, texture, intensity, and margin features, followed by classifiers such as support vector machines, random forests, or k-nearest neighbours. Although these approaches provided diagnostic support, their performance depended heavily on manually designed features. As a result, their generalizability was limited when lesion appearance varied across scanners, acquisition settings, breast densities, and patient populations. Deep learning methods have become popular because they can automatically learn discriminative features from mammographic images without depending on manually designed descriptors.
Convolutional neural networks have been widely used for mammographic image analysis because they can learn local spatial patterns such as edges, texture, lesion margins, and density variations. Investigations have shown that CNN models such as EfficientNet, VGG, DenseNet and ResNet are effective in classifying mammograms and finding abnormalities, as the CNN models are capable of extracting multi-level, hierarchical features from images at varying levels of detail, starting from low-level texture features (e.g., edges) through to high-level semantic representations of the images [
13,
14,
15]. However, due to the limited size of a CNN’s local receptive field, there are limits to the spatial features that can be learned to represent long-range contextual information. This fact becomes particularly important when considering how a radiologist would evaluate a mammogram with a possible lesion; in order to evaluate whether there is a lesion, it is essential to consider both the lesion itself and its immediate environment, including the normal composition of tissue surrounding the lesion, the symmetry of the breast, as well as the normal patterns of breast tissue.
U-Net is a widely used encoder–decoder architecture for biomedical image segmentation [
3,
7]. Its contracting path extracts contextual information, while its expanding path restores spatial resolution for pixel-level localization. Skip connections transfer encoder features to the decoder, which is particularly useful when precise lesion boundary delineation is required. The encoder and decoder components of U-Net include a contracting component, which serves to downsample the image and to capture thorough contextual information at a variety of scales, and a corresponding expanding component, which serves to upsample the image to restore the spatial resolution of the resulting segmented image, for exact localization of objects of interest (e.g., lesions). Additionally, U-Net uses skip connections to transfer spatial feature information learned during the compression (i.e., encoding) to the decompressed version (i.e., decoding) of the images. This ability to transfer spatial feature information is especially critical with respect to medical image segmentation applications where precise boundary delineation is required.
Several U-Net variants have been developed to improve mammogram segmentation performance. U-Net++ employs nested and dense skip connection-based pathways from the encoder to the decoder, which adds to the efficiency of representing semantic differences between the encoder and decoder feature sets [
8]. Attention U-Net integrates attention gates into the existing U-Net architecture to direct the network’s attention to relevant target structures while diminishing the influence of irrelevant background regions [
9]. Attention gates are important for analyzing small lesions, which are frequently present in the background of complex anatomical structures in clinical medical images. The effectiveness of using attention gates within the analysis of mammogram images to concentrate on target (i.e., lesion) regions with heterogeneous contour and size, eliminating the need for the inclusion of additional explicit localization modules within their analysis [
5].
Despite the success of CNN- and U-Net-based methods, breast lesion analysis remains challenging because lesions may appear as irregular masses, diffuse density variations, architectural distortions, or small calcifications. These abnormalities occur at different spatial scales, and a single-scale feature extraction method may fail to capture both fine lesion boundaries and global contextual information. Therefore, multi-scale feature learning has become important in biomedical image analysis. Multi-scale networks combine feature representations from different resolution levels, improving both lesion localization and classification performance in mammograms, especially when lesion size and morphology vary considerably.
Vision Transformers are becoming increasingly popular for medical image analysis because they can model long-range dependencies using self-attention [
10]. In comparison to CNNs, which mainly learn local patterns through convolutions, Transformers model relationships between distant image regions [
11,
12]. This ability is useful in mammography because diagnostic interpretation often requires understanding both the lesion region and the surrounding breast tissue context.
On the other hand, traditional Vision Transformers can be very computationally intensive and require large datasets, as the self-attention mechanism is applied to the entirety of the pixels in the image.
The Swin Transformer overcomes these challenges by implementing a hierarchical Transformer architecture with a shift-window approach. Instead of applying the global self-attention mechanism to all the pixels in the image (like traditional Vision Transformers), Swin Transformers utilize self-attention extracted locally from groups of windows that are shifted relative to the original [
13]. This method allows for effective computation while also allowing for the extraction of multiple scales of representations. The Swin Transformer can be employed as the backbone for a variety of vision tasks, such as classification, detection, and segmentation. This hierarchical design supports efficient feature extraction at multiple resolutions and is therefore suitable for high-resolution medical images such as mammograms [
14].
Many researchers have examined Swin Transformer-based segmentation and classification models for medical imaging. Researchers have found that Swin Transformers provide a viable pathway to achieve hierarchical feature learning through strong contextual modeling, due to the self-attention mechanism. The shift-window approach of the Swin Transformer is advantageous in terms of computational feasibility when applying global attention to high-resolution medical images, especially mammography. Due to the potential to lose local features when applying a pure transformer model, hybrid models that utilize both CNNs and transformers have gained popularity [
14]. This approach combines the strength of the CNN for extracting local features and the strength of the Transformer to model long-range relationships, using self-attention.
Hybrids are particularly applicable to mammographic images. The CNN component provides the ability to extract local features such as texture of the input, edge detail of the input, and margins of lesions, while the transformer component extracts more abstract relationships and semantics of the input. Recent studies examining the segmentation of medical images using the hybrid CNN–Transformer approach have found that this model produces improved representation learning and segmentation performance, especially for data with lesion boundary ambiguity, low contrast, and complex anatomical backgrounds [
15,
16,
17].
Attention-based feature fusion is the other major research direction. Features are typically extracted from many different levels in medical imaging networks, but they are frequently fused using simple addition or concatenation methods. These fusion techniques are computationally cheap but do not allow models to select the most relevant features dynamically. Attention-based fusion techniques and cross-attention mechanisms allow a model to give more weight to diagnostically significant features and reduce the weight assigned to features that are not relevant or are redundant. This is particularly advantageous for breast lesion analyses, as the abnormal areas may be small, difficult to see, and surrounded by significant normal tissue.
Despite the reported advances, the principal model categories involve different trade-offs. CNN and U-Net-based networks effectively preserve local texture and lesion-margin information, but their contextual modelling is constrained by convolutional receptive fields. Pure transformer and Swin Transformer models capture long-range relationships more effectively; however, successive patch merging and spatial downsampling may weaken small-lesion and boundary information. U-shaped transformer models recover spatial detail using decoders and skip connections, but these structures may increase computational demand and transfer redundant low-level features.
Hybrid CNN–Transformer architectures combine local convolutional features with global transformer representations, but many require parallel feature-extraction branches and employ fixed addition or concatenation during fusion. These operations do not explicitly determine whether a particular spatial scale is relevant to a given lesion. Attention-based fusion can partially address this issue, although some methods integrate only selected feature levels or use a full encoder–decoder pathway designed principally for segmentation. Consequently, there remains scope for a framework that adaptively integrates the complete hierarchical Swin feature pyramid and uses the fused representation for both classification and lightweight lesion localization.
Table 2 presents the mammographic and medical-image analysis strategies.
From an architectural perspective, the proposed ICF mechanism differs from commonly used multi-scale integration strategies in three respects. First, conventional U-Net-type skip fusion transfers encoder representations directly to corresponding decoder levels, whereas ICF aligns all four hierarchical Swin stages into a common feature space before fusion. Second, hybrid CNN–Transformer approaches frequently combine independently extracted local and contextual representations through fixed addition or concatenation; in contrast, ICF operates entirely within the hierarchical Swin feature pyramid and recalibrates the contribution of each scale before aggregation. Third, attention-based multi-scale methods can dynamically emphasize informative representations, but several existing approaches operate on selected feature levels or remain embedded within comparatively large encoder–decoder segmentation structures. MSF-Swin-ICFNet instead uses adaptive four-stage cross-scale fusion to generate one shared image-derived representation that supports both classification and lightweight lesion localization. Therefore, the distinction lies not in multi-scale processing alone, but in the sequence of complete hierarchical feature extraction, explicit alignment, adaptive-scale recalibration, weighted fusion, and shared dual-task utilization.
Recent transformer-based breast-imaging studies can be differentiated according to prediction task, input organization, feature-integration strategy, and computational design, the author proposed a multi-stage framework for breast-lesion segmentation and classification [
14], while author also investigated transformer-based explainable breast-lesion segmentation [
15]. The authors had examined multi-view graph- and transformer-based architectures for mammography classification, and Incorporated shape-guided information into transformer-based breast-cancer prediction [
16,
17]. The authors had proposed Swin-DAMFN, in which a Swin Transformer branch for global-context modelling is combined with a CNN-based dual-attention multi-scale fusion branch for local mammographic features [
18,
19,
20,
21,
22]. This architecture demonstrates the value of multi-scale global–local fusion but differs from the present framework because it employs parallel Transformer and CNN pathways rather than adaptive fusion of the complete hierarchical Swin feature pyramid.
Recent segmentation-oriented studies provide additional relevant comparisons. HTU-Net for breast-mass segmentation in mammograms, combining convolutional local representations with Transformer-based global context and a Multi-Scale Cross-Attention Transformer module within an encoder–decoder architecture [
24]. In contrast, the proposed ICF mechanism does not use a parallel CNN–Transformer representation or conventional encoder–decoder skip fusion; instead, all four Swin stages are projected and aligned into a common feature space before adaptive scale weighting. The authors had introduced a multi-scale attention-based Swin Transformer architecture for medical-image segmentation, demonstrating the broader relevance of efficient hierarchical and multi-scale attention mechanisms, although that framework is segmentation-oriented and was not developed specifically for mammographic joint classification and localization [
25].
More recent mammography-specific Swin architectures further demonstrate the rapid development of this area. LAM-CATNet, a Lambda-aware multi-scale cross-attention Swin framework combining Transformer-derived context with U-Net-based local representations for mammographic lesion segmentation and classification [
26]. The authors had investigated a Swin-based framework for analysis of individual mammograms using hierarchical feature learning and optimization of Swin hyperparameters [
27]. These studies confirm that Swin-based hierarchical representation, multi-scale integration, and single-mammogram analysis are active research directions. Therefore, the novelty of MSF-Swin-ICFNet should not be attributed simply to the use of Swin Transformer, multi-scale features, or image-only input. Its specific architectural contribution is the explicit alignment of the complete four-stage Swin hierarchy into a common feature space, adaptive estimation of scale contributions, weighted cross-scale fusion, and use of the resulting shared representation for both benign–malignant classification and lightweight lesion localization without a separate CNN encoder or full U-shaped segmentation decoder [
28].
Because the recent methods differ in dataset composition, prediction endpoint, input organization, architecture, and evaluation protocol, direct numerical ranking across published studies would not constitute a controlled comparison. The present literature comparison is therefore used to establish architectural similarities and differences, while quantitative model comparisons in this study are restricted to architectures reimplemented under the common patient-disjoint experimental protocol.
Overall, the literature does not indicate a single architecture that simultaneously optimizes local boundary preservation, long-range contextual modelling, architectural simplicity, input availability, and joint classification–localization capability. CNN/U-Net approaches retain strong spatial detail but provide comparatively limited long-range modelling; pure Transformer approaches strengthen contextual representation but may weaken fine spatial information through hierarchical downsampling; hybrid CNN–Transformer and attention-based methods improve feature complementarity but may introduce additional branches or fusion complexity; and multimodal approaches can exploit patient-specific information but depend on the consistent availability of additional clinical inputs. These trade-offs motivate the present investigation of an image-derived architecture that adaptively integrates hierarchical Swin features while using a shared representation for classification and lightweight lesion localization.
The proposed design is intended to improve global context modelling while preserving local spatial information from mammographic images. Unlike multimodal or volumetric approaches, the present framework relies only on image-derived Swin Transformer features. The Image-Derived Cross-Scale Fusion module integrates multi-resolution feature maps, while the lightweight boundary-aware mask head supports lesion localization. Therefore, MSF-Swin-ICFNet provides an image-only framework for breast lesion classification and segmentation using CBIS-DDSM and RTM datasets.
A related previous study was proposed SwinCAMF-Net, an explainable multimodal Swin Transformer framework for mammogram analysis [
23]. That model integrated mammographic views, ROI-derived representations and clinical metadata using a multimodal cross-attention fusion module for joint classification and segmentation. While that study demonstrated the value of multimodal fusion, its architecture depends on additional non-image or volumetric information streams. In contrast, the present study focuses on a proposed framework that uses only mammographic image feature representations, thereby addressing a different practical scenario where clinical metadata or ROI-derived inputs may not be consistently available.
Multimodal learning represents an important complementary direction in which imaging-derived representations are integrated with patient-level clinical variables. Such integration can be advantageous when clinical descriptors provide information that is not fully encoded in the image itself. For example, Combined CT-derived deep-learning features with clinical descriptors for predicting pembrolizumab response in advanced non-small-cell lung cancer [
23]. Their patient-level analysis showed that the combined clinical + deep-learning configuration provided better balanced predictive performance than either the deep-learning-only or clinical-only configurations, illustrating that clinical variables can contribute complementary information to image-derived representations. Although the disease, imaging modality, cohort size, and prediction endpoint differ substantially from mammographic lesion classification, the study demonstrates that exclusion of clinical information should not itself be interpreted as an architectural or predictive advantage.
Accordingly, the present study does not argue that image-only modelling is inherently superior to multimodal modelling. Rather, image-only and multimodal approaches address different methodological and deployment settings. Multimodal systems may exploit complementary patient-specific information and potentially improve individualized prediction when sufficiently complete, reliable, and standardized clinical variables are available. Conversely, their performance and portability may be affected by missing variables, heterogeneous recording practices, and differences in the availability or definition of clinical descriptors across institutions. The objective of MSF-Swin-ICFNet is therefore not to replace multimodal approaches, but to investigate a distinct image-derived representation problem: whether the complete hierarchical Swin feature pyramid can be explicitly aligned, adaptively recalibrated, and jointly exploited for classification and lightweight lesion localization. A direct image-only versus image–clinical comparison using identical mammography cohorts, patient partitions, and standardized clinical variables would be valuable future work.
2.1. Research Gap
Despite advances in CNN-, Transformer-, hybrid, and multimodal frameworks for medical-image analysis, an unresolved architectural problem remains in efficiently exploiting the complementary information contained across the complete hierarchical feature pyramid of a Swin Transformer. Shallow stages retain comparatively fine spatial and lesion-boundary information, whereas deeper stages encode progressively more abstract semantic and contextual representations. Existing approaches commonly use the deepest representation, transfer encoder features through decoder skip connections, combine selected scales using fixed addition or concatenation, or employ separate CNN and Transformer branches. Although attention-based fusion can improve feature selection, not all approaches explicitly align the complete four-stage Swin hierarchy and learn the relative contribution of each scale before constructing a shared representation for both classification and lesion localization. Therefore, the primary research gap addressed in this study is not the absence of clinical metadata. Rather, it is the lack of an explicitly formulated image-derived mechanism that performs complete hierarchical feature alignment, adaptive cross-scale recalibration, weighted fusion, and shared utilization of the resulting representation for benign–malignant classification and lightweight lesion localization. The image-only setting defines the scope of the present investigation, while multimodal image–clinical fusion remains a complementary research direction when appropriate clinical information is available.
2.2. Motivation
The study is motivated by the complementary information encoded at different stages of a hierarchical Swin Transformer. Mammographic lesions may contain fine margin, texture, and boundary characteristics together with broader contextual patterns in the surrounding breast tissue, and these characteristics are represented differently across shallow and deep feature levels. An effective fusion mechanism should therefore preserve the contribution of fine spatial representations while also exploiting deeper semantic context and should determine the relative importance of these feature levels adaptively rather than combining them with predetermined equal weights. MSF-Swin-ICFNet was developed to investigate this image-representation problem through cross-scale alignment, adaptive recalibration, and shared classification–localization learning. Its image-only configuration represents the scope of the present experiment rather than a claim that clinical variables lack diagnostic value.
2.3. Proposed Work
This study proposes MSF-Swin-ICFNet, an image-derived framework consisting of hierarchical Swin Transformer feature extraction, adaptive cross-scale feature fusion, a benign–malignant classification head, and a lightweight lesion-localization head. The framework is evaluated using the CBIS-DDSM and RTM mammography datasets through classification, segmentation, patient-grouped cross-validation, computational assessment, comparative experiments, and ablation analysis.
2.4. Objectives of the Work
The primary objective of this work is to investigate whether adaptively fused hierarchical mammographic image features can support reliable benign–malignant classification together with lesion localization in an image-only framework. A secondary objective is to evaluate the proposed design on CBIS-DDSM and RTM using classification and segmentation measures, patient-grouped cross-validation, computational analysis, comparisons with reimplemented architectures, and component-wise ablation experiments.
6. Visualization of Results
This section presents the quantitative and visual evaluation of MSF-Swin-ICFNet on the CBIS-DDSM and RTM mammographic datasets. The analysis includes classification performance, segmentation performance, cross-validation results, ablation analysis, attention visualization, and comparison with existing transformer-based and hybrid models. Visual explanations are used to assess whether the model focuses on clinically relevant lesion regions during prediction. The proposed method generated stable classification and segmentation results on both datasets. The Swin Transformer encoder captured hierarchical image features, while the Multi-Scale Feature Fusion module combined low-level boundary information with high-level semantic representations. The proposed framework enhanced diagnostically relevant regions, improving the model’s ability to classify and localize lesions, especially in cases with irregular shape, low contrast, dense breast tissue, or abnormal tissue patterns.
6.1. Computational Performance of the Model
The computational performance of MSF-Swin-ICFNet was evaluated using an NVIDIA A100 GPU with 40 GB memory and a batch size of 8. The model required an average training time of 10.8 min per epoch, with a total training time of approximately 7.2 h. Peak training memory usage was 31.9 GB, and the average training speed was 25.1 samples per second. During inference, the model required approximately 0.26 s per sample, with an inference memory usage of 5.6 GB and a throughput of 13,846 samples per hour. Each inference required 42.7 GFLOPs. The final model contained 87.2 million total parameters, of which 84.6 million were trainable. The final model size was 312.7 MB, with a feature dimension of 1024. These results indicate that MSF-Swin-ICFNet provides a practical balance between computational complexity and joint diagnostic and lesion-localization performance. Although the model requires a high-memory GPU during training, its inference time is suitable for offline mammographic image analysis and research-level decision-support applications.
6.2. Segmentation Performance
Segmentation performance was evaluated at the mammogram level using images with corresponding lesion masks. The proposed MSF-Swin-ICFNet achieved a Dice coefficient of 0.786 ± 0.041 on CBIS-DDSM and 0.878 ± 0.046 on RTM. The corresponding IoU values were 0.654 ± 0.053 and 0.791 ± 0.052, respectively. Pixel-level precision was 0.822 ± 0.039 for CBIS-DDSM and 0.860 ± 0.041 for RTM, while pixel-level recall was 0.819 ± 0.035 and 0.914 ± 0.043, respectively. The corresponding HD95 values were 3.24 ± 0.89 mm and 2.96 ± 0.88 mm. The two-sided 95% confidence intervals were 0.779–0.793 for CBIS-DDSM Dice and 0.870–0.886 for RTM Dice, with corresponding intervals for the remaining metrics reported in
Table 9.
The segmentation results in
Table 8 show a clear difference in localization behaviour between the two evaluated cohorts, with RTM exhibiting greater region overlap and recall together with somewhat better boundary agreement, whereas CBIS-DDSM represents a comparatively more challenging segmentation setting. This difference should not be attributed directly to the proposed architecture because CBIS-DDSM and RTM differ in cohort origin, acquisition environment, lesion composition, image characteristics, and annotation provenance. The supplementary analysis provides the corresponding patient-disjoint segmentation results and confidence intervals in
Supplementary Table S7.
Several dataset-related mechanisms may contribute to this observation. First, differences in mammography acquisition systems, image-processing characteristics, spatial resolution, contrast, and overall image quality may alter lesion conspicuity and boundary visibility. However, sufficiently detailed acquisition information was not available for a controlled equipment-specific comparison between the two cohorts. Second, segmentation difficulty may vary according to lesion size, morphology, lesion-to-background contrast, and breast composition. Third, reference-annotation characteristics can substantially influence overlap- and boundary-based segmentation measures. The RTM masks were supplied by the dataset provider, but information concerning the original delineation procedure, number and expertise of annotators, inter-observer variability, and consensus process was unavailable. Consequently, differences in annotation consistency cannot be excluded as a contributor to the observed performance gap.
Consequently, the higher RTM segmentation scores should not be interpreted as evidence that the model is intrinsically more effective on clinically acquired mammograms. A dedicated cross-dataset analysis would be required to determine the origin of the observed performance difference. Such an analysis should stratify segmentation performance according to lesion size, morphology, contrast, breast density, and image-quality characteristics and should, where possible, examine annotation variability and boundary agreement. These analyses would help determine whether the difference between CBIS-DDSM and RTM arises primarily from lesion characteristics, acquisition quality, reference-mask properties, or interactions among these factors. Therefore, the present cross-dataset difference is reported as an empirical observation rather than a causal finding.
Figure 3 compares the Dice coefficient, IoU, precision, recall, and Hausdorff distance across both datasets. Higher Dice, IoU, precision, and recall values indicate better segmentation quality, while lower Hausdorff distance indicates improved boundary accuracy.
6.3. Mammogram-Level Classification Performance
The classification performance reported in this section was calculated at the individual-mammogram level. Although each mammogram was processed and scored independently, the test images were obtained from patient-disjoint cohorts; therefore, no patient contributing a test mammogram was represented in the training or validation subsets. Patient-level probability aggregation, patient-level classification metrics, and patient-level confusion matrices are reported separately in the
Supplementary Materials. Patient-level ROC-AUC results and their confidence intervals are provided in
Supplementary Table S8. The results in
Table 10 indicate that MSF-Swin-ICFNet maintained a favorable balance between malignant-case sensitivity and benign-case specificity within the evaluated patient-disjoint cohorts. Nevertheless, differences between CBIS-DDSM and RTM should not be interpreted as evidence that one cohort is intrinsically easier or that the model generalizes more effectively to clinical data. Apparent classification performance can be affected by cohort composition, lesion prevalence, lesion conspicuity, breast composition, acquisition environment, and other dataset-specific characteristics. Therefore, the cross-dataset results are interpreted as descriptive evidence of performance within the evaluated cohorts rather than as proof of domain-independent clinical robustness. Complementary patient-level results obtained by aggregating mammogram predictions are provided in the
Supplementary Materials.
On the CBIS-DDSM patient-disjoint test cohort, MSF-Swin-ICFNet achieved a mammogram-level accuracy of 0.949, precision of 0.944, recall of 0.919, specificity of 0.971, F1-score of 0.931, ROC-AUC of 0.985, and PR-AUC of 0.949. On the RTM patient-disjoint test cohort, MSF-Swin-ICFNet achieved a mammogram-level accuracy of 0.968, precision of 0.953, recall of 0.959, specificity of 0.974, F1-score of 0.956, ROC-AUC of 0.992, and PR-AUC of 0.970. These results were calculated from 239 CBIS-DDSM mammograms and 1489 RTM mammograms belonging to patients who were not represented in the training or validation partitions. The complete mammogram-level metric set is also provided in
Supplementary Table S5. Patient-level results obtained by aggregating mammogram probabilities are reported separately in
Supplementary Table S4, and the corresponding performance comparison is illustrated in
Supplementary Figure S1. Thus, the main manuscript results represent mammogram-level performance, whereas the supplementary analysis provides complementary patient-level evaluation. A supplementary visual comparison of mammogram-level classification performance, including ROC and precision–recall analyses, is provided in
Supplementary Figure S2. The corresponding numerical patient-level confusion matrices are provided in
Supplementary Table S6. Visual representations of these patient-level confusion matrices are provided in
Supplementary Figure S3. Complementary patient-level classification results obtained by aggregating mammogram-level probabilities are reported in
Supplementary Table S4, and the corresponding performance comparison is illustrated in
Supplementary Figure S1. The complete mammogram-level metric set is also provided in
Supplementary Table S5, with a visual comparison presented in
Supplementary Figure S2. The corresponding numerical patient-level confusion matrices are provided in
Supplementary Table S6, with their visual representations shown in
Supplementary Figure S3.
Figure 4 presents a visual comparison of the classification metrics, including accuracy, precision, recall, F1-score, AUC-ROC, and AUC-PR, across CBIS-DDSM five-fold cross-validation, CBIS-DDSM test, and RTM test settings. The figure highlights the consistent and improved performance of the proposed model across different evaluation settings. Patient-level ROC-AUC results and their confidence intervals are provided in
Supplementary Table S8, with the corresponding ROC curves shown in
Supplementary Figure S4. Patient-level PR-AUC values are reported in
Supplementary Tables S8 and S9 and the corresponding precision–recall curves are shown in
Supplementary Figure S5.
6.4. F1-Score Comparison Between CBIS-DDSM and RTM
The F1-score comparison of the proposed MSF-Swin-ICFNet model with existing deep learning models is presented in
Table 11. The F1-score was selected because it provides a balanced evaluation of precision and recall, which is particularly important in breast cancer classification where both false positives and false negatives have clinical significance.
The F1-score comparison in
Table 10 shows that MSF-Swin-ICFNet achieved the highest F1-score among the models included in the present comparison. The previously published SwinCAMF-Net reported F1-scores of 0.902 on CBIS-DDSM and 0.944 on RTM, whereas MSF-Swin-ICFNet achieved 0.931 and 0.958, respectively. However, the SwinCAMF-Net results were taken from the previous publication and were not re-evaluated using the patient-disjoint protocol adopted in the present study. Therefore, these values are presented as contextual comparisons rather than as results from a controlled head-to-head evaluation.
The above
Figure 5 presents the F1-score performance of ResNet, DenseNet, Vision Transformer, Medical Transformer, Swin-T v2, and the proposed MSF-Swin-ICFNet model.
6.5. Mammogram-Level Performance Under Patient-Grouped Cross-Validation
Five-fold patient-grouped cross-validation was conducted within the RTM development cohort comprising 487 patients. The independent 86-patient test cohort remained locked and was excluded from cross-validation. All mammograms associated with a patient remained within the same fold. Approximately 97–98 patients served as validation patients in each fold, while approximately 389–390 patients were available for training in the corresponding folds. Consequently, the model was evaluated on individual images without permitting mammograms from the same patient to appear in both the training and validation portions of a fold. The complete patient and fold composition used for the five-fold patient-grouped cross-validation is provided in
Supplementary Table S3. Five-fold cross-validation was conducted on the CBIS-DDSM dataset to evaluate the stability of the proposed MSF-Swin-ICFNet model across different data partitions. Unlike a single train–test split, cross-validation provides a more reliable estimate of model generalization by evaluating performance across multiple folds.
The CBIS-DDSM five-fold cross-validation results showed an accuracy of 0.881 ± 0.012, precision of 0.864 ± 0.019, recall of 0.867 ± 0.018, and F1-score of 0.858 ± 0.012. The corresponding AUC-ROC and AUC-PR values were 0.931 ± 0.015 and 0.907 ± 0.011, respectively. These results indicate stable discrimination between benign and malignant cases across different validation folds. Five-fold patient-grouped cross-validation was conducted on the CBIS-DDSM development cohort to assess the stability of the proposed model across different patient partitions. The RTM dataset was evaluated using its own patient-disjoint training, validation, and held-out test subsets and therefore represents an additional clinical-cohort evaluation rather than a fully external validation cohort. The patient-level confusion matrices were verified against the corresponding test-cohort composition. For CBIS-DDSM, the four cells sum to 134 patients (62 TN + 3 FP + 6 FN + 63 TP), comprising 65 benign and 69 malignant patients. For RTM, the four cells sum to 86 patients (44 TN + 2 FP + 2 FN + 38 TP), comprising 46 benign and 40 malignant patients. Thus, the confusion-matrix counts are consistent with the stated patient-level test cohorts. Accordingly, RTM performance should be interpreted as evidence of model performance on a clinically acquired dataset with different image characteristics, but not as independent external clinical validation and is therefore reported separately in
Table 12.
Figure 6 presents the mean cross-validation scores for accuracy, precision, recall, F1-score, AUC-ROC, and AUC-PR. The results show that the RTM dataset achieved higher values across all major performance metrics, while the CBIS-DDSM results remained stable with narrow confidence intervals. This supports the robustness of the proposed model across different mammographic datasets.
Figure 7 shows the discrimination ability of the proposed MSF-Swin-ICFNet model on the independent patient-disjoint test cohorts, with mammogram-level ROC-AUC values of 0.985 for (a) CBIS-DDSM and 0.992 for (b) RTM.
Figure 8 shows the mammogram-level precision–recall performance of the proposed MSF-Swin-ICFNet model on the independent patient-disjoint test cohorts. The model achieved PR-AUC values of 0.949 for CBIS-DDSM and 0.970 for RTM.
Figure 9 and
Figure 10 show the patient-level confusion matrices of MSF-Swin-ICFNet on the independent patient-disjoint test cohorts. For CBIS-DDSM, the test cohort contained 134 patients, comprising 65 benign and 69 malignant patients. The resulting confusion matrix contained 62 true negatives, 3 false positives, 6 false negatives, and 63 true positives. For RTM, the test cohort contained 86 patients, comprising 46 benign and 40 malignant patients, with 44 true negatives, 2 false positives, 2 false negatives, and 38 true positives. These confusion matrices correspond to the patient-level classification analysis reported in the
Supplementary Materials. The corresponding patient-level PR-AUC values are reported in
Supplementary Table S9. The corresponding patient-level precision–recall curves are shown in
Supplementary Figure S5.
6.6. Ablation Study
An ablation study was conducted to evaluate the contribution of each major component in the proposed MSF-Swin-ICFNet framework. The full model was compared with reduced variants by removing the proposed module, cross-scale recalibration, feature refinement layers, and the boundary-aware mask head. The purpose of this experiment was to verify whether each image-derived component contributes meaningfully to classification and lesion mask prediction performance.
The ablation study in
Table 13 demonstrates that each major component of MSF-Swin-ICFNet contributes to the overall classification and lesion-localization performance. Removing Image-Derived Cross-Scale Fusion reduced the CBIS-DDSM accuracy, F1-score, AUC-ROC, and Dice by 4.3%, 6.3%, 4.0%, and 5.6%, respectively, while the corresponding reductions for RTM were 5.6%, 6.1%, 2.0%, and 3.6%. Removing cross-scale recalibration resulted in further reductions, particularly in F1-score and Dice, indicating that adaptive weighting of the aligned feature representations contributes to the fused representation. The removal of feature-stack refinement produced larger reductions in both classification and segmentation performance, while removing the boundary-aware mask head primarily affected lesion-localization performance. The Swin Encoder Only configuration produced the lowest overall performance across both datasets, providing an additional comparison against using the backbone without the proposed fusion and refinement components. Overall, the ablation results support the contribution of the individual architectural components to the joint classification and lesion-localization framework.
Figure 11 presents the ablation results for the CBIS-DDSM dataset. The full MSF-Swin-ICFNet model shows better accuracy, F1-score, AUC-ROC, and Dice coefficient than the reduced variants, indicating that each module contributes to classification and segmentation performance.
Figure 12 shows the ablation results on the RTM dataset. Removing Image-Derived Cross-Scale Fusion (ICF), cross-scale recalibration, feature refinement layers, or boundary-aware mask head reduces model performance, which confirms the importance of these components in improving feature fusion, lesion representation, and boundary localization.
Figure 13 compares the ablation trends across CBIS-DDSM and RTM datasets. The trend shows that the complete model performs consistently better than all reduced configurations, while the basic Swin Transformer gives the lowest overall performance.
6.7. Comparison with Previous SwinCAMF-Net Study
The previously published SwinCAMF-Net and the proposed MSF-Swin-ICFNet are both based on Swin Transformer features but differ in their input modalities and fusion strategies [
23]. SwinCAMF-Net integrates multi-view mammograms, 3D ROI volumes, and clinical metadata through a cross-attention fusion module, followed by classification and segmentation branches. In contrast, MSF-Swin-ICFNet focuses on hierarchical image features and cross-scale feature integration without requiring additional volumetric and clinical input streams.
The previously published SwinCAMF-Net achieved F1-scores of 0.902 on CBIS-DDSM and 0.944 on RTM, while MSF-Swin-ICFNet achieved 0.931 and 0.958, respectively. For the Dice/AUC comparison, SwinCAMF-Net reported 0.795/0.957 on CBIS-DDSM and 0.931/0.977 on RTM, whereas MSF-Swin-ICFNet achieved 0.786/0.985 and 0.878/0.992, respectively. Thus, MSF-Swin-ICFNet provides higher reported classification AUC values, whereas SwinCAMF-Net provides higher reported segmentation Dice values, particularly on the RTM dataset.
SwinCAMF-Net has the advantage of incorporating complementary volumetric and clinical information, which can provide additional contextual information for lesion analysis. Its multimodal design, however, requires additional input modalities and associated processing. MSF-Swin-ICFNet uses image-derived features and therefore provides a simpler input configuration while emphasizing hierarchical cross-scale feature integration. Nevertheless, the lower Dice values obtained by MSF-Swin-ICFNet indicate that its lesion-localization performance remains below that reported for SwinCAMF-Net. Therefore, the two approaches represent different architectural trade-offs rather than one model being universally superior to the other.
Since the SwinCAMF-Net results were reproduced from the previous publication and were not re-evaluated using the exact patient-disjoint data partitions and experimental protocol of the present study, this comparison should be interpreted as contextual rather than as a definitive head-to-head evaluation.
6.8. Visualization of Model Performance
The visualizations of the proposed MSF-Swin-ICFNet model enable a detailed understanding of the regions of interest that relate to lesions in the process of breast cancer classification and segmentation, based on how the network focuses on the relevant areas of the image. Grad-CAM maps were generated to identify image regions that contributed most strongly to the model predictions. Correct benign and malignant predictions were examined together with false-positive and false-negative cases to assess whether the model attended to clinically meaningful lesion regions.
In TP+ classifications, the attention maps demonstrate a strong and focal activation over the regions of the image that correspond to the suspicious lesion(s), indicating that the model correctly focused on diagnostically higher abnormal areas of the image. In TN- classifications, the level of activation occurs in a less focused and distributed fashion across the imaged normal tissue representing that the model did not detect malignant patterns. In FP+ classifications, the attention areas will activate near benign structures which may visually appear similar to malignant abnormal findings such as dense tissue or atypical contour. In FN- classifications, activation of attention occurs in a weak or scattered manner and does not adequately demonstrate focus of attention on the actual lesion region.
Overall, the Grad-CAM visualizations in
Figure 14 indicate that MSF-Swin-ICFNet focuses on clinically important areas in mammograms. Therefore, these findings lend support to the interpretability of this model, while providing information about successful and unsuccessful predictions. Furthermore, these visualizations are useful for interpretability in the model, identify its limitations, and direct the improvement of its architectural features. The results of this visual analysis are consistent with the CBIS-DDSM and RTM datasets, which report the model’s performance with respect to classification and segmentation.
Figure 14 shows original mammograms and their corresponding attention heatmaps for true positive, true negative, false positive, and false negative predictions. Strong localized activations indicate that the model had focused its activation on the lesion correctly, while weak or scattered activations signal the possibility of failure to correctly predict lesions.
6.9. Comparison with State-of-the-Art Transformer Models
To provide a controlled comparison with representative deep-learning, transformer-based, and hybrid architectures, the comparator models reimplemented in the present study were evaluated under the same experimental protocol used for MSF-Swin-ICFNet. The previously published SwinCAMF-Net results are included separately as a contextual reference and were not re-evaluated under the present protocol. Specifically, ResNet, DenseNet, Vision Transformer, Medical Transformer, Swin-T v2, ViT+U-Net, TransUNet, EfficientNet-B3, Swin-UNet, MedT, UNETR, and CoTr were trained and evaluated using the same patient-disjoint CBIS-DDSM and RTM data partitions, preprocessing pipeline, augmentation policy, and evaluation metrics adopted for the proposed model. The training, validation, and testing cohorts were therefore identical across the compared architectures. For MSF-Swin-ICFNet, the final mammogram-level classification ROC-AUC values were 0.985 on CBIS-DDSM and 0.992 on RTM, while the corresponding lesion-localization Dice scores were 0.786 and 0.878, respectively. These results indicate high classification discrimination within the evaluated patient-disjoint cohorts, while the lower Dice score on CBIS-DDSM indicates comparatively more challenging lesion localization. Consequently, the comparison demonstrates the joint classification–localization capability of the proposed framework rather than universal superiority across every segmentation endpoint. All reimplemented models were trained using the same patient-disjoint cohorts and common experimental protocol; consequently, the comparative results reported in
Table 11 and
Table 14 were generated under consistent data-processing and evaluation conditions.
The controlled comparison therefore indicates a trade-off rather than uniform dominance across all evaluated endpoints. The proposed framework provides a common image-derived representation for both classification and lesion localization, but several dedicated segmentation architectures achieve higher overlap scores under particular conditions. This observation is especially relevant for CBIS-DDSM, where the lightweight localization pathway does not outperform all segmentation-specific comparator architectures. Conversely, the classification results indicate that adaptive hierarchical cross-scale fusion provides a strong representation for benign–malignant discrimination within the evaluated cohorts.
These results support the usefulness of the proposed architecture as a joint classification–localization framework while also identifying lesion-boundary delineation as an area requiring further improvement.
Figure 15 therefore illustrates the comparative trade-off between classification discrimination and segmentation performance rather than evidence of universal superiority of the proposed method. Because all models were evaluated using identical patient-disjoint cohorts, preprocessing procedures, and evaluation metrics, the comparison reduces confounding associated with differences in dataset partitioning and experimental conditions. The observed performance differences therefore provide a more controlled assessment of the contribution of the proposed image-derived cross-scale fusion and lightweight lesion-localization strategy.
Figure 15 compares the proposed MSF-Swin-ICFNet model with existing transformer-based and hybrid models using segmentation Dice score and classification AUC. The proposed model achieved a Dice score of 0.786 and ROC-AUC of 0.985 on CBIS-DDSM, and a Dice score of 0.878 and ROC-AUC of 0.992 on RTM. These results indicate strong classification performance across both datasets, while the lower Dice score on CBIS-DDSM suggests more challenging segmentation conditions.
Table 14 provides a controlled comparison of MSF-Swin-ICFNet with representative reimplemented transformer, hybrid, and deep-learning architectures under common patient-disjoint data partitions and evaluation conditions. The comparison reveals different behaviour for classification and segmentation. MSF-Swin-ICFNet achieved strong classification discrimination relative to the evaluated comparator architectures, whereas its segmentation performance was not uniformly superior across all methods and datasets. In particular, the CBIS-DDSM results indicate that the proposed lightweight localization pathway remains sensitive to challenging lesion boundaries and dataset-specific annotation characteristics, while the RTM results show more competitive segmentation behaviour.
These findings suggest that the principal contribution of MSF-Swin-ICFNet lies in obtaining a shared hierarchical representation that supports both benign–malignant classification and lightweight lesion localization rather than in claiming universal superiority for pixel-level segmentation. Cross-dataset differences should also be interpreted cautiously because CBIS-DDSM and RTM differ in acquisition environment, lesion characteristics, image properties, and annotation provenance. The available data do not permit these influences to be isolated individually; therefore, the comparative results are interpreted as performance under the present controlled experimental protocol rather than as evidence of identical behaviour across external clinical populations.
The proposed model achieved a Dice score of 0.786 and ROC-AUC of 0.985 on CBIS-DDSM, and a Dice score of 0.878 and ROC-AUC of 0.992 on RTM. These results indicate strong classification performance across both datasets, while the lower Dice score on CBIS-DDSM suggests comparatively more challenging lesion-localization conditions. The comparison should therefore be interpreted as a trade-off between classification discrimination, lesion localization, and computational complexity rather than as evidence of universal superiority across all endpoints.
6.10. Discussions and Limitations
There are several potential sources of dataset bias that should be considered when interpreting the present results. First, CBIS-DDSM and RTM represent different data-acquisition environments. CBIS-DDSM is a curated public benchmark, whereas RTM is a retrospectively collected clinical mammography cohort. Differences in acquisition systems, image quality, lesion prevalence, preprocessing characteristics, and annotation practices may introduce domain or acquisition bias. Because sufficiently detailed and harmonized equipment- and acquisition-level information was not available across the evaluated cohorts, the individual contribution of scanner, detector, acquisition-protocol, and image-processing differences could not be quantified in the present study. Although a standardized preprocessing and normalization strategy was applied, these procedures cannot completely remove variability originating from different imaging environments. Therefore, the performance observed on the evaluated datasets may not directly generalize to mammograms obtained using different imaging systems, acquisition protocols, or clinical institutions.
Second, population and demographic bias cannot be fully assessed, particularly for the RTM cohort. Detailed demographic characteristics, including age distribution, ethnicity, menopausal status, and breast-density distribution, were not available for subgroup analysis. Therefore, the current results do not establish whether model performance is consistent across different demographic or breast-composition groups. The higher segmentation performance observed on RTM relative to CBIS-DDSM could not be attributed to a specific source because lesion morphology, lesion size, breast density, image-quality characteristics, and annotation variability were not systematically matched or stratified across the two datasets. Therefore, future cross-dataset analyses should examine these factors explicitly before drawing conclusions regarding dataset-specific segmentation difficulty or model generalizability.
Third, potential selection and annotation bias should be considered for RTM segmentation. Although the complete RTM dataset contained 10,070 mammograms, lesion masks were available for only 4863 images (48.3%). Consequently, the mask-bearing subset used for segmentation may not fully represent the lesion characteristics and difficulty distribution of the complete RTM cohort. In addition, the original annotation protocol, number of annotators, and their level of expertise were not available for independent verification. The RTM masks were therefore treated as provider-supplied reference annotations rather than independently verified clinical ground-truth delineations.
Fourth, multiple mammograms may originate from the same patient, introducing within-patient correlation among image-level observations. To minimize information leakage, all mammograms belonging to an individual patient were retained within a single training, validation, or testing partition. Nevertheless, the primary performance estimates reported in the manuscript are mammogram-level measures and should not be interpreted as statistically independent patient observations. Complementary patient-level analyses are provided in the
Supplementary Materials.
An important limitation of the present study is the absence of fully independent multicentre external validation. Although the evaluation included both the public CBIS-DDSM benchmark and the retrospectively collected RTM clinical mammography cohort, these datasets were incorporated within the model-development and evaluation framework and therefore do not constitute independent multicentre external validation. Consequently, the reported results primarily demonstrate performance within the characteristics of the evaluated cohorts and should not be interpreted as evidence that the model will maintain the same performance across unseen clinical institutions. Variations in mammography systems, acquisition protocols, image quality, patient demographics, breast-density distributions, disease prevalence, and annotation practices may influence model behaviour when deployed in a new clinical environment.
This limitation is particularly important when considering the potential clinical applicability of an AI-based mammography system. Before clinical translation, the finalized model should be frozen and evaluated without retraining, fine-tuning, or threshold optimization on independent patient cohorts obtained from institutions that were not involved in model development. Such external validation should include heterogeneous imaging systems and patient populations and should assess both mammogram-level and patient-level classification performance, together with lesion-localization performance where reliable reference annotations are available. Prospective multicentre studies and radiologist reader evaluations would provide further evidence regarding the robustness, generalizability, and potential clinical utility of the proposed framework. Therefore, the current findings should be regarded as promising retrospective evidence rather than definitive evidence of clinical generalizability. Clinical translation also requires evaluation beyond discrimination metrics alone. A high AUC does not establish an appropriate clinical operating threshold or demonstrate benefit to radiologist decision-making. Future external studies should therefore assess calibration, clinically relevant sensitivity–specificity trade-offs, and performance across lesion and breast-density subgroups whenever reliable metadata are available. Radiologist reader studies should determine whether the classification probability and lesion-localization output improve sensitivity, specificity, interpretation time, or diagnostic confidence relative to unaided assessment. Attention should also be given to discordant cases in which a high malignancy probability is associated with weak or spatially implausible localization. Consequently, the present model should be regarded as a research-stage decision-support framework until reproducible clinical benefit is demonstrated in independent prospective settings.
Future multicentre validation should assess both mammogram-level and patient-level performance in independent clinical populations. Of the 10,070 RTM mammograms, 4863 images (48.3%) were accompanied by provider-supplied lesion masks. The complete RTM cohort contained 10,070 mammograms and was eligible for image-level classification. Of these, 4863 mammograms (48.3%) had corresponding provider-supplied lesion masks and were eligible for segmentation analysis. Patient-level partitioning was performed before constructing the mask-bearing segmentation subsets, and each mammogram and corresponding mask inherited the partition assignment of its source patient. Consequently, no image–mask pair belonging to one patient could appear in more than one partition. RTM mammograms without lesion masks contributed to classification analysis but did not contribute to the segmentation loss or segmentation evaluation. Additional information regarding the annotation procedure and the number and experience of the mask annotators was not available in the released documentation. The RTM masks were therefore treated as dataset-provided reference annotations rather than independently verified provider-supplied reference truth annotations. In addition, detailed RTM demographic information and institution-specific provenance were not available for subgroup analysis, and the original mask-annotation protocol and annotator qualifications could not be independently verified.
Although the study included the clinically acquired RTM cohort in addition to the public CBIS-DDSM benchmark, RTM was not used as a purely external validation cohort because patient-disjoint RTM subsets were used during model development and evaluation. Therefore, the present results should not be interpreted as independent external clinical validation. A rigorous external validation study would require freezing the final model after development on the source cohort and evaluating it, without retraining, fine-tuning, threshold optimization, or model selection, on mammograms obtained from an entirely independent institution. Such multicentre external validation remains an important direction for future work.