Dataset Condensation Atlas

Application · Trajectory matching

BTM

Geometric Characterisation and Structured Trajectory Surrogates for Clinical Dataset Condensation

Pafue Christy Nganjimi, Andrew Soltan, Danielle Belgrave, Lei Clifton, David Clifton, Anshul Thakur

arXiv 2026 · first public 2026-04-23 · arXiv 2604.21638

paper ↗catalogued✓ full text read

In one paragraph

Gives a geometric account of why trajectory matching is hard to supervise with a small fixed synthetic set: such a set can only reproduce a limited, low-rank span of the parameter changes real SGD training induces, so a spectrally broad supervision signal creates a representability bottleneck. Proposes Bezier Trajectory Matching (BTM), which replaces SGD expert trajectories with quadratic Bezier surrogates between initial and final model states, reducing trajectory storage and better matching what a fixed synthetic set can represent. Reports BTM matching or improving on standard trajectory matching across five clinical tabular-EHR and time-series datasets (three NHS emergency-department cohorts, eICU and MIMIC-III), with the largest gains at low prevalence and low synthetic-data budgets.

Where it sits

Abstract (verbatim from arXiv)

Dataset condensation constructs compact synthetic datasets that retain the training utility of large real-world datasets, enabling efficient model development and potentially supporting downstream research in governed domains such as healthcare. Trajectory matching (TM) is a widely used condensation approach that supervises synthetic data using changes in model parameters observed during training on real data, yet the structure of this supervision signal remains poorly understood. In this paper, we provide a geometric characterisation of trajectory matching, showing that a fixed synthetic dataset can only reproduce a limited span of such training-induced parameter changes. When the resulting supervision signal is spectrally broad, this creates a conditional representability bottleneck. Motivated by this mismatch, we propose Bezier Trajectory Matching (BTM), which replaces SGD trajectories with quadratic Bezier trajectory surrogates between initial and final model states. These surrogates are optimised to reduce average loss along the path while replacing broad SGD-derived supervision with a more structured, lower-rank signal that is better aligned with the optimisation constraints of a fixed synthetic dataset, and they substantially reduce trajectory storage. Experiments on five clinical datasets demonstrate that BTM consistently matches or improves upon standard trajectory matching, with the largest gains in low-prevalence and low-synthetic-budget settings. These results indicate that effective trajectory matching depends on structuring the supervision signal rather than reproducing stochastic optimisation paths.

BibTeX (generated; prefer the venue's official entry)
@article{nganjimi2026geometric,
  title   = {Geometric Characterisation and Structured Trajectory Surrogates for Clinical Dataset Condensation},
  author  = {Pafue Christy Nganjimi and Andrew Soltan and Danielle Belgrave and Lei Clifton and David Clifton and Anshul Thakur},
  journal = {arXiv preprint arXiv:2604.21638},
  year    = {2026}
}

Nearby in Trajectory matching

2026-03

PTM-ST — Multimodal Dataset Distillation via Phased Teacher Models

Shengbin Guo, Hang Zhao, Senqiao Yang et al. · ICLR 2026notableVision–languagepaper ↗code ↗

2026-01

AMD — Asynchronous Matching with Dynamic Sampling for Multimodal Dataset Distillation

Ding Qi, Jian Li, Shuguang Dou et al. · ICLR 2026notableVision–languagepaper ↗

2025-05

RepBlend — Beyond Modality Collapse: Representations Blending for Multimodal Dataset Distillation

Xin Zhang, Ziruo Zhang, Jiawei Du et al. · NeurIPS 2025notableVision–languagepaper ↗

2024-10

MKDT — Dataset Distillation via Knowledge Distillation: Towards Efficient Self-Supervised Pre-Training of Deep Networks

Siddharth Joshi, Jiayi Ni, Baharan Mirzasoleiman · ICLR 2025notablePre-training & transferpaper ↗code ↗

2024-08

LTDD — Distilling Long-tailed Datasets

Zhenghao Zhao, Haoxuan Wang, Yuzhang Shang et al. · CVPR 2025notablepaper ↗code ↗