Search papers, labs, and topics across Lattice.
This paper introduces OrganLens, a novel approach for organ-specific representation learning in CT imaging that utilizes self-supervision to condition a shared encoder on organ identity. By employing organ-specific distillation and anatomy-mask supervision, the model generates 11 distinct organ representations without the need for external segmentation masks. Evaluations demonstrate significant improvements in clinical metrics, such as a rise in cardiomegaly detection AUROC from 0.910 to 0.953 for heart representations and a 14.2% enhancement in lung cancer mortality prediction for lung representations.
OrganLens achieves unprecedented organ-specific representation learning in CT scans, significantly boosting diagnostic accuracy without requiring external segmentation.
A CT examination captures multiple organs, but many biomedical questions concern abnormalities, prognosis, or longitudinal change in a specific organ. These questions require a separate representation for each organ within the same CT volume. Existing CT foundation models commonly produce a single volume-level representation, while recent anatomy-aware methods either encode pre-separated organ volumes or explicitly disentangle images into organ token groups. The former may remove clinically relevant surrounding context, while the latter does not condition a shared encoder on a selected organ before its features are formed. We introduce OrganLens for organ-specific representation learning through self-supervision. An organ identity conditions a shared CT encoder, while organ-specific distillation and anatomy-mask supervision shape features for anatomy-weighted pooling into organ-specific representations. At inference, the shared model produces 11 organ-specific representations without external segmentation masks. We evaluate OrganLens on CT-RATE, RAD-ChestCT, INSPECT, and NLST across diverse acquisitions and downstream evaluations. Relative to CT-pretrained DINOv2, heart representations raise CT-RATE cardiomegaly AUROC from 0.910 to 0.953, while lung representations improve the Harrell C-index for NLST lung-cancer mortality by 14.2\%. The global representation reaches INSPECT Recall@10 of 33.09\% and 32.04\% for text-to-image and image-to-text retrieval, respectively. Across organ-related tasks, anatomically matched representations provide stronger task-relevant signal, while the global representation retains broad utility. OrganLens offers a scalable approach to organ-specific CT representation learning with a shared encoder. More broadly, it provides the medical research community with a reusable framework for studying organ-specific disease across cohorts and clinical endpoints.
UW