X-WIN: Building Chest Radiograph World Model via Predictive Sensing

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Yang, Zefan, Wang, Ge, Hendler, James, Kalra, Mannudeep K., Yan, Pingkun
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866914385896669184
author Yang, Zefan
Wang, Ge
Hendler, James
Kalra, Mannudeep K.
Yan, Pingkun
author_facet Yang, Zefan
Wang, Ge
Hendler, James
Kalra, Mannudeep K.
Yan, Pingkun
contents Chest X-ray radiography (CXR) is an essential medical imaging technique for disease diagnosis. However, as 2D projectional images, CXRs are limited by structural superposition and hence fail to capture 3D anatomies. This limitation makes representation learning and disease diagnosis challenging. To address this challenge, we propose a novel CXR world model named X-WIN, which distills volumetric knowledge from chest computed tomography (CT) by learning to predict its 2D projections in latent space. The core idea is that a world model with internalized knowledge of 3D anatomical structure can predict CXRs under various transformations in 3D space. During projection prediction, we introduce an affinity-guided contrastive alignment loss that leverages mutual similarities to capture rich, correlated information across projections from the same volume. To improve model adaptability, we incorporate real CXRs into training through masked image modeling and employ a domain classifier to encourage statistically similar representations for real and simulated CXRs. Comprehensive experiments show that X-WIN outperforms existing foundation models on diverse downstream tasks using linear probing and few-shot fine-tuning. X-WIN also demonstrates the ability to render 2D projections for reconstructing a 3D CT volume.
format Preprint
id arxiv_https___arxiv_org_abs_2511_14918
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle X-WIN: Building Chest Radiograph World Model via Predictive Sensing
Yang, Zefan
Wang, Ge
Hendler, James
Kalra, Mannudeep K.
Yan, Pingkun
Computer Vision and Pattern Recognition
Chest X-ray radiography (CXR) is an essential medical imaging technique for disease diagnosis. However, as 2D projectional images, CXRs are limited by structural superposition and hence fail to capture 3D anatomies. This limitation makes representation learning and disease diagnosis challenging. To address this challenge, we propose a novel CXR world model named X-WIN, which distills volumetric knowledge from chest computed tomography (CT) by learning to predict its 2D projections in latent space. The core idea is that a world model with internalized knowledge of 3D anatomical structure can predict CXRs under various transformations in 3D space. During projection prediction, we introduce an affinity-guided contrastive alignment loss that leverages mutual similarities to capture rich, correlated information across projections from the same volume. To improve model adaptability, we incorporate real CXRs into training through masked image modeling and employ a domain classifier to encourage statistically similar representations for real and simulated CXRs. Comprehensive experiments show that X-WIN outperforms existing foundation models on diverse downstream tasks using linear probing and few-shot fine-tuning. X-WIN also demonstrates the ability to render 2D projections for reconstructing a 3D CT volume.
title X-WIN: Building Chest Radiograph World Model via Predictive Sensing
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2511.14918