Track papers across labs, topics, and trends.
Search papers, labs, and topics across Lattice.
This study introduces a federated multi-task learning framework for bladder tumor segmentation and muscle invasion classification using a hybrid CNN-Transformer architecture, addressing the challenge of data centralization across clinical centers. The Swin Hybrid model integrates a ResNet-34 for local texture analysis and a Swin-Tiny Transformer for capturing global anatomical context, while a segmentation-guided classification mechanism enhances MIBC prediction accuracy. Results from the FedBCa dataset reveal that the proposed method achieves a DSC of 0.8100 and a patient-level AUC of 0.8931 under federated training, showcasing its effectiveness in multi-institutional settings without compromising patient data privacy.
Joint segmentation and classification of bladder tumors can be achieved across institutions without centralizing sensitive patient data, leveraging federated learning for robust performance.
Accurate bladder tumor segmentation and assessment of mus- cle invasion from T2-weighted MRI are important for treatment plan- ning, but developing robust models across institutions is challenging be- cause patient data cannot be centrally pooled and imaging characteristics vary across scanners and acquisition protocols. We propose a federated multi-task learning framework for joint bladder tumor segmentation and MIBC/NMIBC classification across four clinical centers. The proposed Swin Hybrid model combines a ResNet-34 branch for local texture and boundary information with a Swin-Tiny Transformer for global anatomi- cal context. A segmentation-guided classification mechanism further uses tumor localization information to support MIBC prediction. We also investigate several augmentation strategies under both centralized and federated training to improve robustness to multi-center variability. Ex- periments on the FedBCa dataset show that the Swin Hybrid provides the best overall balance between segmentation and classification among the evaluated architectures. Under federated training, Geo+Elastic aug- mentation achieved a DSC of 0.8100 and a patient-level AUC of 0.8931, yielding the highest combined score of 0.8474. These results demonstrate that joint segmentation and classification can be effectively performed across multiple institutions using federated training without centralizing patient data.