mlsanityData to 14 Oct 2026

Geometric Characterisation and Structured Trajectory Surrogates for Clinical Dataset Condensation

arXiv 2604.21638 ↗First posted 23 Apr 2026cs.LGcited by 0

Not on the map. Why?

Abstract (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.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 18 of its 29 references that are in the map. First posted on arXiv 23 Apr 2026.

Builds on

18 of 29 references.

References · 18

  1. 2018An Empirical Evaluation of Generic Convolutional and Recurrent Networks for Sequence Modelingcited by 1,476
  2. 2018Averaging Weights Leads to Wider Optima and Better Generalizationcited by 1,209
  3. 2018Loss Surfaces, Mode Connectivity, and Fast Ensembling of DNNscited by 588
  4. 2020Dataset Condensation with Gradient Matchingcited by 516
  5. 2022Dataset Distillation by Matching Training Trajectoriescited by 442
  6. 2021Dataset Condensation with Distribution Matchingcited by 365
  7. 2018Essentially No Barriers in Neural Network Energy Landscapecited by 330
  8. 2021Dataset Condensation with Differentiable Siamese Augmentationcited by 291
  9. 2020Dataset Meta-Learning from Kernel Ridge-Regressioncited by 230
  10. 2021Dataset Distillation with Infinitely Wide Convolutional Networkscited by 218
  11. 2022Scaling Up Dataset Distillation to ImageNet-1K with Constant Memorycited by 181
  12. 2022Minimizing the Accumulated Trajectory Error to Improve Dataset Distillationcited by 142
6 more
  1. 2023Squeeze, Recover and Relabel: Dataset Condensation at ImageNet Scale From A New Perspectivecited by 138
  2. 2023Towards Lossless Dataset Distillation via Difficulty-Aligned Trajectory Matchingcited by 118
  3. 2023On the Diversity and Realism of Distilled Dataset: An Efficient Dataset Distillation Paradigmcited by 109
  4. 2020Optimizing Mode Connectivity via Neuron Alignmentcited by 86
  5. 2024Towards Stable and Storage-efficient Dataset Distillation: Matching Convexified Trajectorycited by 21
  6. 2023Data Encoding For Healthcare Data Democratisation and Information Leakage Preventioncited by 3

Built on by

Not cited yet.