Mask-Guided Attention Regulation for Anatomically Consistent Counterfactual CXR Synthesis

• URL: http://arxiv.org/abs/2603.04130v1
• Date: Wed, 04 Mar 2026 14:44:41 GMT
• Title: Mask-Guided Attention Regulation for Anatomically Consistent Counterfactual CXR Synthesis
• Abstract summary: We present an inference-time attention regulation framework for reliable counterfactual CXR synthesis.<n>An anatomy-aware attention regularization module gates self-attention and anatomy-token cross-attention with organ masks.<n>A pathology-guided module enhances pathology-token cross-attention within target lung regions during early denoising.
• Score: 28.400345187461085
• License: http://arxiv.org/licenses/nonexclusive-distrib/1.0/
• Abstract: Counterfactual generation for chest X-rays (CXR) aims to simulate plausible pathological changes while preserving patient-specific anatomy. However, diffusion-based editing methods often suffer from structural drift, where stable anatomical semantics propagate globally through attention and distort non-target regions, and unstable pathology expression, since subtle and localized lesions induce weak and noisy conditioning signals. We present an inference-time attention regulation framework for reliable counterfactual CXR synthesis. An anatomy-aware attention regularization module gates self-attention and anatomy-token cross-attention with organ masks, confining structural interactions to anatomical ROIs and reducing unintended distortions. A pathology-guided module enhances pathology-token cross-attention within target lung regions during early denoising and performs lightweight latent corrections driven by an attention-concentration energy, enabling controllable lesion localization and extent. Extensive evaluations on CXR datasets show improved anatomical consistency and more precise, controllable pathological edits compared with standard diffusion editing, supporting localized counterfactual analysis and data augmentation for downstream tasks.

Related papers

• AnaDiffusion: Anatomically CompositionalLatent Diffusion for Controllable 3D Brain MRI Generation [58.56402940400457]
  3D brain MRI generation has made significant advances in medical imaging, simulation, and controllable anatomical analysis.<n>We introduce AnaDiffusion, an anatomically compositional latent diffusion framework that factorizes the generation process into distinct, anatomically meaningful regions.<n>AnaDiffusion produces both explicit part assets and a globally coherent volume, enabling controllable part editing.
  arXiv  Detail & Related papers  (2026-08-24T09:16:42Z)
• Stain-Aware Wavelet Regularization for Instant Adversarial Purification in Histopathology [12.803436797824268]
  We propose Stain-Aware Wavelet Regularization (SAWR) to disentangle adversarial perturbations from diagnostic structural information.<n>When integrated into an instant purification framework, SAWR improves adversarial robustness by up to 10.69% over the baseline approach.
  arXiv  Detail & Related papers  (2026-06-07T17:32:24Z)
• Anatomy-Informed Deep Learning for Abdominal Aortic Aneurysm Segmentation [0.0]
  We propose an anatomy-aware segmentation framework that integrates organ exclusion masks into the training process.<n>These masks encode explicit anatomical priors by identifying non-vascular organs and penalizing aneurysm predictions within these regions.<n>Despite being trained on a relatively small dataset, the anatomy-aware model achieves anatomically high accuracy, substantially reduces false positives, and improves boundary consistency compared to a standard U-Net baseline.
  arXiv  Detail & Related papers  (2026-04-11T18:27:35Z)
• Structure-constrained Language-informed Diffusion Model for Unpaired Low-dose Computed Tomography Angiography Reconstruction [72.80209358480424]
  overdose of iodinated contrast media (ICM) can cause kidney damage and life-threatening allergic reactions.<n>Deep learning methods can generate CT images of normal-dose ICM from low-dose ICM, reducing the required dose.<n>We propose a Structure-constrained Language-informed Diffusion Model (SLDM) that integrates structural synergy and spatial intelligence.
  arXiv  Detail & Related papers  (2026-01-28T06:54:06Z)
• PathoSyn: Imaging-Pathology MRI Synthesis via Disentangled Deviation Diffusion [11.223559964746705]
  We present PathoSyn, a unified generative framework for Magnetic Resonance Imaging (MRI) image synthesis.<n>PathoSyn reformulates imaging-pathology as a disentangled additive deviation on a stable anatomical manifold.<n>We show that PathoSyn significantly outperforms holistic diffusion and mask-conditioned baselines in both perceptual realism and anatomical fidelity.
  arXiv  Detail & Related papers  (2025-12-29T01:13:50Z)
• AnyCXR: Human Anatomy Segmentation of Chest X-ray at Any Acquisition Position using Multi-stage Domain Randomized Synthetic Data with Imperfect Annotations and Conditional Joint Annotation Regularization Learning [0.0]
  We propose AnyCXR, a unified framework that enables generalizable multi-organ segmentation across arbitrary CXR projection angles.<n>We trained AnyCXR entirely on synthetic data, providing accurate delineation of 54 anatomical structures in PA, lateral, and oblique views.<n>Results demonstrate that AnyCXR establishes a scalable and reliable foundation for anatomy-aware CXR analysis.
  arXiv  Detail & Related papers  (2025-12-19T06:27:14Z)
• Self-Supervised Anatomical Consistency Learning for Vision-Grounded Medical Report Generation [61.350584471060756]
  Vision-grounded medical report generation aims to produce clinically accurate descriptions of medical images.<n>We propose Self-Supervised Anatomical Consistency Learning (SS-ACL) to align generated reports with corresponding anatomical regions.<n>SS-ACL constructs a hierarchical anatomical graph inspired by the invariant top-down inclusion structure of human anatomy.
  arXiv  Detail & Related papers  (2025-09-30T08:59:06Z)
• Automated Labeling of Intracranial Arteries with Uncertainty Quantification Using Deep Learning [2.6279333406008476]
  We present a deep learning-based framework for automated artery labeling from 3D Time-of-Flight Magnetic Resonance Angiography (3D ToF-MRA)<n>Our framework offers a scalable, accurate, and uncertainty-aware solution for automated cerebrovascular labeling, supporting downstream hemodynamic analysis and facilitating clinical integration.
  arXiv  Detail & Related papers  (2025-09-22T12:57:21Z)
• Prompt-Guided Patch UNet-VAE with Adversarial Supervision for Adrenal Gland Segmentation in Computed Tomography Medical Images [0.3437656066916039]
  Small abdominal organs, such as the adrenal glands in CT imaging, remains a persistent challenge due to severe class imbalance, poor spatial context, and limited annotated data.<n>We propose a unified framework that combines variational reconstruction, supervised segmentation, and adversarial patch-based feedback to address these limitations in a principled and scalable manner.<n>Our findings highlight the effectiveness of hybrid generative-discriminative training regimes for small-organ segmentation and provide new insights into balancing realism, diversity, and anatomical consistency in data-scarce scenarios.
  arXiv  Detail & Related papers  (2025-09-03T10:18:06Z)
• CLAIM: Clinically-Guided LGE Augmentation for Realistic and Diverse Myocardial Scar Synthesis and Segmentation [3.052913696182197]
  We introduce CLAIM: textbfClinically-Guided textbfLGE textbfAugmentation for Realtextbfiiyocardial Scar Synthesis and framework.<n>At its core is the SMILE module, which conditions a diffusion-based generator on the clinically adopted AHA 17-segment model to synthesize images with anatomically consistent and spatially diverse scar patterns.<n> Experimental results show that CLAIM produces anatomically coherent scar patterns and achieves higher Dice similarity with real scar distributions compared to baseline models.
  arXiv  Detail & Related papers  (2025-06-18T15:21:34Z)
• RadFabric: Agentic AI System with Reasoning Capability for Radiology [61.25593938175618]
  RadFabric is a multi agent, multimodal reasoning framework that unifies visual and textual analysis for comprehensive CXR interpretation.<n>System employs specialized CXR agents for pathology detection, an Anatomical Interpretation Agent to map visual findings to precise anatomical structures, and a Reasoning Agent powered by large multimodal reasoning models to synthesize visual, anatomical, and clinical data into transparent and evidence based diagnoses.
  arXiv  Detail & Related papers  (2025-06-17T03:10:33Z)
• Diffusion Models with Implicit Guidance for Medical Anomaly Detection [13.161402789616004]
  Temporal Harmonization for Optimal Restoration (THOR) aims to preserve the integrity of healthy tissue in areas unaffected by pathology. Relative evaluations show THOR surpasses existing diffusion-based methods in detecting and segmenting anomalies in brain MRIs and wrist X-rays.
  arXiv  Detail & Related papers  (2024-03-13T12:26:55Z)
• Cross-Modal Causal Intervention for Medical Report Generation [107.76649943399168]
  Radiology Report Generation (RRG) is essential for computer-aided diagnosis and medication guidance.<n> generating accurate lesion descriptions remains challenging due to spurious correlations from visual-linguistic biases.<n>We propose a two-stage framework named CrossModal Causal Representation Learning (CMCRL)<n> Experiments on IU-Xray and MIMIC-CXR show that our CMCRL pipeline significantly outperforms state-of-the-art methods.
  arXiv  Detail & Related papers  (2023-03-16T07:23:55Z)
• Structure-aware registration network for liver DCE-CT images [50.28546654316009]
  We propose a novel structure-aware registration method by incorporating structural information of related organs with segmentation-guided deep registration network. Our proposed method can achieve higher registration accuracy and preserve anatomical structure more effectively than state-of-the-art methods.
  arXiv  Detail & Related papers  (2023-03-08T14:08:56Z)
• Fuzzy Attention Neural Network to Tackle Discontinuity in Airway Segmentation [67.19443246236048]
  Airway segmentation is crucial for the examination, diagnosis, and prognosis of lung diseases. Some small-sized airway branches (e.g., bronchus and terminaloles) significantly aggravate the difficulty of automatic segmentation. This paper presents an efficient method for airway segmentation, comprising a novel fuzzy attention neural network and a comprehensive loss function.
  arXiv  Detail & Related papers  (2022-09-05T16:38:13Z)

This list is automatically generated from the titles and abstracts of the papers in this site.

This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.