Beyond Independent Frames: Latent Attention Masked Autoencoders for Multi-View Echocardiography

Simon Böhi, Irene Cannistraci, Sergio Muñoz Gonzalez, Moritz Vandenhirtz, Sonia Laguna, Samuel Ruiperez-Campillo, Max Krähenmann, Andrea Agostini, Ece Ozkan, Thomas M. Sutter, Julia E Vogt

Keywords: Echocardiography, Masked Autoencoder, Multi-view, Representation Learning, Self-Supervised Learning, Transformer, Transfer Learning

TL;DR: We introduce LAMAE, a latent-attention masked autoencoder that aggregates variable-length, multi-view echocardiography videos in latent space to learn robust and transferable cardiac representations.

Abstract:

Echocardiography is a widely used modality for cardiac assessment due to its non-invasive and cost-effective nature, but the sparse and heterogeneous spatiotemporal views of the heart pose distinct challenges.
Existing masked autoencoder (MAE) approaches typically process images or short clips independently, failing to capture the inherent multi-view structure required for coherent cardiac representation.
We introduce Latent Attention Masked Autoencoder (LAMAE), a foundation model architecture tailored to the multi-view nature of medical imaging.
LAMAE augments the standard MAE with a latent attention module that enables information exchange across frames and views directly in latent space.
This allows the model to aggregate variable-length sequences and distinct views, reconstructing a holistic representation of cardiac function from partial observations.
We pretrain LAMAE on MIMIC-IV-ECHO, a large-scale, uncurated dataset reflecting real-world clinical variability.
To the best of our knowledge, we present the first results for predicting ICD-10 codes from MIMIC-IV-ECHO videos.
Furthermore, we empirically demonstrate that representations learned from adult data transfer effectively to pediatric cohorts despite substantial anatomical differences.
These results provide evidence that incorporating structural priors, such as multi-view attention, yields significantly more robust and transferable representations.

Submission Number: 45