From Baseline to Follow-Up: Counterfactual Spine DXA Image Synthesis in UK Biobank Using a Causal Hierarchical Variational Autoencoder

· Source: Computer Vision and Pattern Recognition · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Computer Vision & Pattern Recognition, Health & Medical Research · Depth: Expert, quick

Summary

A metadata-conditioned causal hierarchical variational autoencoder (CHVAE) has been developed for causally consistent generation of anteroposterior (AP) spine DXA images. This model addresses the challenge of learning controllable and interpretable factor-specific anatomical variation in large-scale skeletal assessments. Trained on 3,743 raw AP spine scans from the UK Biobank's first imaging visit, the CHVAE is conditioned on basic participant attributes and lumbar morphometry. Its causal consistency was evaluated in a baseline-to-follow-up setting using an abduction--action--prediction (AAP) framework. This involved abducting latent variables from baseline images, intervening on age to match repeat-imaging values, and comparing the resulting counterfactual follow-up morphometry with observed measurements. The results demonstrate strong absolute-level agreement for key vertebral morphometry variables under age intervention, supporting the synthesis of anatomically plausible DXA images.

Key takeaway

For AI Scientists developing medical image synthesis models, this CHVAE approach offers a robust framework for generating causally consistent DXA images. You should consider integrating metadata conditioning and the abduction--action--prediction (AAP) evaluation strategy to ensure anatomical plausibility and interpretability in your synthetic data. This can significantly enhance the utility of generated images for studying disease progression or intervention effects.

Key insights

The CHVAE model synthesizes anatomically plausible, causally consistent spine DXA images by intervening on age, addressing anatomical variation challenges.

Principles

Method

The CHVAE is trained on UK Biobank DXA scans, conditioned on participant attributes and lumbar morphometry. It uses abduction--action--prediction (AAP) to intervene on age and compare generated counterfactuals with observed follow-up data.

In practice

Topics

Best for: AI Scientist, Research Scientist, Computer Vision Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Computer Vision and Pattern Recognition.