Search papers, labs, and topics across Lattice.
NVIDIAChung-AngKyung Hee UniversityOUTTASamsung Changwon HospitalSKKUSNUJul 13, 2026arXiv:2607.11533This study addresses the challenge of preoperative prediction of perineural invasion (PNI) in cholangiocarcinoma using a novel adaptive routing mechanism within a diffusion transformer framework. By integrating denoising-based class scoring with transformer architectures, the authors enhance the model's ability to capture subtle imaging features while significantly reducing computational overhead. The proposed method achieves an area under the curve (AUC) of 0.731 with a computational cost of 257.57 GFLOPs, marking a notable improvement in efficiency and predictive performance.
Adaptive routing in transformer-based models can drastically enhance the efficiency of PNI prediction while maintaining high accuracy in capturing subtle imaging features.
Perineural invasion (PNI) is a critical prognostic factor in cholangiocarcinoma. However, its preoperative prediction from magnetic resonance imaging (MRI) remains challenging due to subtle imaging features that extend beyond tumor boundaries into surrounding regions. Conventional convolutional neural networks are limited in capturing long-range spatial dependencies. Transformer-based architectures improve global modeling of volumetric MRI by aggregating spatially distributed contextual cues, yet capturing subtle and noise-sensitive patterns in peritumoral regions remains challenging. Diffusion-based classifiers offer an alternative formulation by leveraging denoising-based class scoring to better capture such subtle patterns. However, these approaches introduce substantial computational overhead due to the combination of transformer-based modeling and iterative denoising processes. To address these challenges, we formulate PNI prediction as a diffusion-based classification problem and implement the denoising network using a transformer-based representation. To improve computational efficiency, we introduce adaptive routing across attention heads, spatial tokens, and MLP width. Experimental results demonstrate that the proposed approach achieves an AUC of 0.731 with 257.57 GFLOPs.
