Conceptio › Archive › arXiv CS
arXiv CSopen access

Geometric Characterisation and Structured Trajectory Surrogates for Clinical Dataset Condensation

2026 · arxiv_cs
arXiv CS · Papers · License: Open Access · 2026
Open Source ↗Direct PDF ↓
neural-networks
machine learning, deep learning, neural networks

Geometric Characterisation and Structured Trajectory Surrogates for Clinical Dataset Condensation∗ Pafue Christy Nganjimi

[email protected]

Department of Engineering Science University of Oxford, UK

arXiv:2604.21638v1 [cs.LG] 23 Apr 2026

Andrew Soltan

[email protected]

Nuffield Department of Primary Care University of Oxford, UK

Danielle Belgrave

[email protected]

GlaxoSmithKline London, UK

Lei Clifton

[email protected]

Nuffield Department of Primary Care University of Oxford, UK

David Clifton

[email protected]

Department of Engineering Science University of Oxford, UK

Anshul Thakur

[email protected]

Department of Engineering Science University of Oxford, UK

Abstract 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 Bézier Trajectory Matching (BTM), which replaces SGD trajectories with quadratic Bézier trajectory surrogates between initial and final model states. These surrogates are optimised to reduce average loss along the path while replacing broad SGDderived 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.

∗ An earlier version of this work appeared as a preprint.

1

4 3

θ2

2

0.8 0.6

1 0.4

0 θt

−1

0.2

−2 −3

−2.5

0.0

2.5

5.0

7.5

0.0

θ1

(a) Parameter-space trajectories

Average Training Loss

SGD Trajectory Mode-connected Path

0.7

0

10

Epochs

20

30

40

50

SGD Trajectory Mode-connected Path

1.0

Trajectory-matching Loss

1.0

θ0

Loss

5

0.6 0.5 0.4 0.3

TM with SGD Teacher Trajectories TM with Mode-connected Surrogates

0.9

0.8

0.7

0.6 0.0

0.2

0.4

0.6

0.8

Interpolation t

(b) Training loss profiles

1.0

0

200

400

600

800

1000

Iteration (x10)

(c) Trajectory-matching optimisation

Figure 1: Illustrative comparison between raw SGD teacher trajectories and Bézier trajectory surrogates used in BTM. (a) Between the same initial and final model states, raw SGD can follow jagged, directionally variable paths, whereas a Bézier surrogate (mode connected path) provides a smoother low-complexity connection. (b) The two forms of supervision also differ functionally. SGD training exhibits irregular epoch-to-epoch loss behaviour, while the Bézier surrogate defines a smooth loss profile between the same endpoints. (c) In trajectory matching, replacing raw SGD segments with structured surrogate supervision can yield a more stable optimisation objective and faster convergence.

1

INTRODUCTION

Modern machine learning systems increasingly rely on large-scale datasets (Pastorino et al., 2019), yet storing, sharing, and repeatedly training on such datasets remains computationally and operationally expensive. In clinical settings, this challenge is compounded by strict governance and access constraints on electronic health records (EHRs) (Thakur et al., 2024; Shabani, 2019), which limit data sharing, restrict multi-centre collaboration, and can make external validation across diverse real-world cohorts more difficult. Dataset condensation (Wang et al., 2018; Zhao & Bilen, 2021a) addresses this problem by constructing compact synthetic datasets that retain the training utility of the original data. In governed domains, such condensed datasets may support downstream methodological research and model development where direct data access is restricted. However, they do not by themselves provide formal privacy guarantees, and their safe deployment would require combining them with differential privacy. Trajectory Matching (TM) (Cazenavette et al., 2022; Du et al., 2023; Guo et al., 2024) is a widely used approach to dataset condensation. Rather than matching gradients or feature statistics at a single point, TM supervises synthetic data using changes in model parameters observed during training on real data. By transferring optimisation dynamics into synthetic datasets, TM provides a direct mechanism for encoding task-relevant structure. Despite strong empirical performance, however, progress in TM has largely relied on empirical heuristics—such as trajectory regularisation, selecting which parts of the training trajectory to match, smoothing (Du et al., 2023; Guo et al., 2024; Zhong et al., 2025), and scalability improvements (Cui et al., 2023)—without a principled understanding of the supervision signal being matched. This leaves a basic question unresolved. Which aspects of real-data training trajectories can a fixed synthetic dataset actually reproduce? In TM, supervision is conveyed through changes in model parameters observed during training on real data, but the structure of this signal and its representational demands are not well understood. Without such an understanding, it remains unclear why trajectory matching performs well in some settings, degrades in others, or how its supervision should be structured to make condensation more effective. In this work, we answer this question through a geometric characterisation of trajectory matching. We show that, for any fixed synthetic dataset, the trajectory matching residual is lower bounded by how well the parameter changes induced by training on real data can be approximated within the gradient span reachable from that dataset. This leads to a conditional representability bottleneck. When the resulting supervision 2

signal is spectrally broad relative to the reachable span, some components necessarily lie outside what the synthetic dataset can reproduce, and a non-zero matching error remains unavoidable. We then study this mismatch empirically for the supervision signal induced by standard SGD training. We find that the parameter changes used by TM often exhibit broad spectral structure, with non-negligible mass distributed across many weakly aligned directions. This effect appears across stochastic mini-batching, random initialisations, and distinct trained solutions, although we do not attempt to isolate the contribution of each source. The mismatch is especially pronounced in low-prevalence settings, where updates associated with the minority class are sparser and less consistently reinforced during training. As a result, standard TM may require a fixed synthetic dataset to reproduce a supervision signal that is broader than its reachable span, offering a plausible explanation for weaker performance in rare-event clinical prediction. To mitigate this limitation, we propose Bézier Trajectory Matching (BTM), which replaces discrete SGD trajectories with task-optimised quadratic Bézier surrogates connecting the initial and final model states. The approach is inspired by mode connectivity (Garipov et al., 2018; Draxler et al., 2018; Thakur et al., 2025), but our setting differs from the classical case because one endpoint is a random initialisation rather than a second converged solution. A quadratic Bézier curve provides a simple nonlinear path between the endpoints through a single learned control point, allowing the surrogate to capture curvature without storing a long sequence of SGD iterates. In practice, this yields smoother start-to-solution paths whose loss decreases more consistently from initialisation to solution. Compared with standard SGD trajectories, the resulting supervision signal is more structured and has lower effective rank, making it better aligned with the representational constraints of a fixed synthetic dataset while also substantially reducing trajectory storage. Figure 1 provides an intuitive comparison between raw SGD supervision and the structured Bézier surrogates used in BTM. The figure illustrates the smoother parameter-space geometry and loss profile of the surrogate paths, together with their effect on trajectory-matching optimisation. We evaluate BTM on five real-world clinical datasets spanning tabular and time-series modalities. Across these tasks, BTM matches or outperforms prior baselines, with the largest gains in low-prevalence and low-synthetic-budget settings, where identifying positive cases reliably is especially important. The main contributions of this paper are summarised as follows: • Geometric characterisation of trajectory matching: We formulate TM for a fixed synthetic dataset as a constrained subspace-approximation problem and show that its residual is lower bounded by the projection error of real-data parameter changes onto the dataset’s reachable gradient span. This reveals a conditional representability bottleneck when that span is limited. • Empirical analysis of standard trajectory supervision: We show that supervision induced by standard SGD trajectories is often spectrally broad, especially in low-prevalence settings, helping to explain the mismatch between real-data training dynamics and what compact synthetic datasets can reproduce. • Bézier Trajectory Matching: We introduce BTM, which replaces stored SGD trajectories with task-optimised quadratic Bézier surrogates, yielding a more structured supervision signal and a substantially more compact trajectory representation. The remainder of the paper is organised as follows. Section 2 reviews related work, Section 3 introduces background on dataset condensation and trajectory matching, Section 4 develops the geometric analysis, Section 5 introduces BTM, Section 6 describes the experimental setup, and Section 7 presents the empirical results.

2

RELATED WORK

Dataset Condensation: Dataset condensation methods differ primarily in the training signal used to optimise synthetic data. Gradient-based approaches match gradients induced by real and synthetic batches, with later extensions incorporating augmentation-aware objectives (Zhao & Bilen, 2021a;b). Feature- and distribution-based methods instead align representations of real and synthetic data in embedding space, for example through feature alignment or distribution matching (Zhao & Bilen, 2023; Wang et al., 2022). 3

Kernel-based approaches provide more principled closed-form or function-space objectives, but are often less scalable to larger datasets and architectures (Nguyen et al., 2021; 2022). Other scalable or decoupled approaches prioritise efficiency, realism, or data reconstruction rather than explicitly modelling optimisation dynamics (Kanagavelu et al., 2024; Yin et al., 2023; Sun et al., 2023). Trajectory-Based Condensation: Trajectory matching (Cazenavette et al., 2022) is particularly relevant to our work because it supervises synthetic data using parameter displacements observed along real training trajectories. Subsequent work has improved this paradigm through flatter or regularised trajectories (Du et al., 2023), stage- or difficulty-aware selection of which parts of the trajectory to match (Guo et al., 2024), and memory-efficient scaling to larger datasets such as ImageNet (Cui et al., 2023). More recently, Zhong et al. (Zhong et al., 2025) proposed matching convexified trajectories to improve stability, convergence speed, and storage efficiency. Their convexified teacher trajectory provides smoother guidance, but it is not explicitly designed to remain in a low-loss region of the parameter space. As a result, in highly non-convex landscapes, smoothing by convexification may trade trajectory stability against fidelity to the original optimisation dynamics. Mode Connectivity: Mode connectivity studies low-loss curves between trained models (Garipov et al., 2018; Draxler et al., 2018), revealing substantial geometric structure in neural-network parameter space. Related work has shown that these curves can be exploited for ensembling and wider optima (Izmailov et al., 2018), and that their optimisation can benefit from explicitly accounting for weight-space symmetries (Tatro et al., 2020). Recent work has also used related mode-connectivity ideas in clinical machine learning settings (Thakur et al., 2025). Our setting differs from the classical mode-connectivity case because one endpoint is a random initialisation rather than a second converged solution. We therefore use the same geometric machinery to learn a smoother descent surrogate, rather than to claim a classical low-loss connection between two minima. Positioning of Our Work: This paper sits at the intersection of trajectory-based condensation and geometric modelling of weight-space curves. Rather than proposing another heuristic modification of stored SGD trajectories, we study the representational structure of the supervision signal induced by trajectory matching. This yields a geometric account of when trajectory supervision is limited by the reachable gradient span of a fixed synthetic dataset, and motivates Bézier trajectory surrogates as a more structured alternative to raw SGD trajectories. In contrast to prior trajectory-based methods, including recent approaches based on convexified teacher trajectories (Zhong et al., 2025), our work asks which aspects of trajectory-induced supervision are reproducible by a fixed synthetic dataset. Whereas convexified trajectories primarily smooth teacher dynamics for improved stability and storage efficiency, BTM replaces each teacher trajectory with a task-optimised quadratic Bézier surrogate motivated by this representability bottleneck, yielding a supervision signal that is more structured and better aligned with what a fixed synthetic dataset can reproduce.

3

Background

3.1

Dataset Condensation Problem

Dataset condensation aims to represent the task-relevant training signal of a large dataset D with a much smaller synthetic dataset D̃, such that models trained on D̃ achieve comparable generalisation performance. Formally, let |D|

D = {(xi , yi )}i=1 ,

|D̃|

D̃ = {(x̃j , ỹj )}j=1 ,

|D̃| ≪ |D|,

and let θ ∈ Θ = Rp denote model parameters. Given an initialisation θ̂0 ∼ P and a fixed training procedure θ̂N = Train(θ̂0 , D̃), dataset condensation seeks a synthetic dataset satisfying   D̃⋆ ∈ arg min Eθ̂0 ∼P Lval (θ̂N ) s.t. θ̂N = Train(θ̂0 , D̃),

(1)

D̃

where Lval denotes a validation loss. The challenge is that D̃ must induce training dynamics that generalise well despite being orders of magnitude smaller than D. 4

3.2

Trajectory Matching

A prominent approach to dataset condensation is trajectory matching, which uses a teacher optimisation trajectory as supervision for learning synthetic data. Let {θτ }Tτ=0 ⊂ Θ denote the parameter sequence obtained by training a teacher model on D. We consider a collection of index pairs {(sk , ek )}K k=1 , with 0 ≤ sk < ek ≤ T , defining the endpoints of K trajectory segments. Each segment induces a teacher displacement ∆k := θek − θsk . To match segment k, a student model is initialised at θ̂0 := θsk and trained on D̃ for N inner steps. Under the first-order, no-momentum update used in standard trajectory matching implementations, the student evolves as θ̂n+1 = θ̂n − ηs gn , gn = ∇θ ℓ(θ̂n ; Bn ), n = 0, . . . , N − 1, (2) where ηs is the student learning rate, Bn ⊂ D̃ is a synthetic mini-batch, and ℓ denotes the mini-batch loss. This yields the student displacement δk := θ̂N − θsk . The standard per-segment trajectory matching objective is 2

LTM (sk , ek ; D̃) := θ̂N − θek

2

2

= ∥δk − ∆k ∥2 .

(3)

Trajectory matching can therefore be viewed as learning a synthetic dataset whose induced optimisation steps reproduce teacher displacements in parameter space. In the next section, we introduce our geometric analysis of this objective and show that the student can only realise displacements lying in a restricted span determined by the gradients accessible from the synthetic data.

4

A Geometric View of Trajectory Matching

This section develops a geometric analysis of trajectory matching. For a fixed synthetic dataset and inner-loop optimiser, the student cannot realise arbitrary teacher displacements; it can move only along directions generated by gradients accessible from D̃. This induces a constrained subspace of parameter space that determines which teacher updates are representable. Two consequences follow. For an individual segment, the trajectory matching loss is lower bounded by the component of the teacher displacement lying outside the student’s reachable span. Across segments, this leads to a rank-based bottleneck showing that broad-spectrum teacher supervision cannot, in general, be reproduced within a low-dimensional reachable span. These results explain why raw SGD trajectories can be difficult to match with compact synthetic datasets and motivate structured surrogate supervision. 4.1

Reachable Gradient Span

To formalise the representational constraint imposed by the synthetic inner loop, we define the reachable gradient span. −1 Definition 1 (Reachable gradient span). Let {gn }N n=0 be the gradients in equation 2 for segment k. The reachable gradient span for segment k is Gk = span{g0 , . . . , gN −1 } ⊂ Rp .

(4)

Under the first-order, no-momentum update in equation 2, the student displacement satisfies δk = −ηs

N −1 X

gn ,

n=0

and therefore δk ∈ Gk . For a fixed segment, trajectory matching is therefore not an unconstrained regression problem in parameter space. The student can approximate the teacher displacement ∆k only by vectors lying in the subspace Gk . 5

4.2

Projection Lower Bound

The subspace constraint above leads directly to a geometric lower bound on the trajectory matching objective. Theorem 1 (Projection lower bound). Fix a segment k with endpoint indices (sk , ek ), let PGk denote the orthogonal projector onto Gk , and let I denote the identity operator on Rp . Then LTM (sk , ek ; D̃) ≥

2

(I − PGk )∆k 2 .

(5)

Moreover, the minimiser of min ∥v − ∆k ∥22

v∈Gk

is PGk ∆k . Thus, the bound is tight as a best-in-subspace approximation statement, although a fixed synthetic dataset need not realise this minimiser. Proof. Section B.1 of the appendix provides the complete proof. Remark 1 (Best-in-span versus realised displacement). Theorem 1 characterises the best approximation to ∆k available within the reachable span Gk . The realised student displacement δk = −ηs

N −1 X

gn

n=0

need not coincide with this optimum for a fixed synthetic dataset. The result should therefore be interpreted as a geometric lower bound rather than as an attainability guarantee. Theorem 1 shows that any component of the teacher displacement orthogonal to the student’s reachable span produces irreducible approximation error. Mismatch is therefore unavoidable whenever the teacher supervision places substantial mass outside the directions that the synthetic inner loop can generate. 4.3

A Rank Bottleneck for Shared Synthetic Supervision

The single-segment view above extends naturally to the full collection of teacher displacements supervised by a shared synthetic dataset. Let A := [∆1 , . . . , ∆K ] ∈ Rp×K ,

rA := rank(A),

(6)

denote the matrix of teacher displacements, and define the global reachable span induced by D̃ as G := span{gn(k) : k = 1, . . . , K, n = 0, . . . , N − 1},

r := dim(G).

(7)

For a fixed synthetic dataset, trajectory matching can therefore be viewed as approximating the displacement matrix A within an r-dimensional subspace G. This yields a global representability bottleneck. Theorem 2 (Rank bottleneck under an r-dimensional reachable span). Let A = [∆1 , . . . , ∆K ] ∈ Rp×K be the teacher displacement matrix, let G ⊂ Rp be a subspace of dimension r with orthogonal projector PG , let I (k) denote the identity operator on Rp , and define LTM := LTM (sk , ek ; D̃). Then K

1 X (k) 1 2 LTM ≥ (I − PG )A F . K K

(8)

k=1

Moreover, among all r-dimensional subspaces G ⊂ Rp , the minimum residual is rA 1 X 1 2 (I − PG )A F = σj (A)2 , K j=r+1 dim(G)=r K

min

where σj (A) denotes the j-th singular value of A. 6

(9)

Proof. Section B.2 of the appendix provides the complete proof. Remark 2. The displacements {∆k }K k=1 may be drawn from one or multiple teacher trajectories (Cazenavette et al., 2022). The analysis depends only on the resulting collection of displacement vectors and applies in either case. Remark 3 (Conditional nature of the bottleneck). Theorem 2 is conditional on the effective dimension r = dim(G) induced by a fixed synthetic dataset, inner-loop horizon, and model class. It does not assert a universal, architecture-independent upper bound on r. Instead, it quantifies the residual once an r-dimensional reachable span is given. Theorem 2 identifies a fundamental limitation of trajectory matching under constrained synthetic supervision. When substantial spectral mass of A lies outside the effective reachable span, a positive residual remains unavoidable. The difficulty of trajectory matching is therefore governed not only by the nominal rank of A, but more generally by its spectrum and effective rank relative to the student’s reachable span. These results suggest that effective trajectory supervision should be concentrated in a small number of coherent directions. The next section introduces Bézier Trajectory Matching, which is designed to impose exactly this kind of structured supervision.

5

Bézier Trajectory Matching

The rank bottleneck above suggests that effective supervision should preserve functionally meaningful progress while avoiding diffuse high-rank variability. Bézier Trajectory Matching (BTM) addresses this limitation by replacing stochastic optimisation trajectories with learnable quadratic Bézier curves that act as structured low-rank surrogates. These surrogate paths are optimised to reduce average loss along the path, suppress unnecessary high-loss excursions, and empirically encourage a smoother, typically decreasing expected loss profile from initialisation to solution. The induced displacement directions are therefore confined to a low-dimensional subspace, yielding supervision that is better aligned with the constraints of student optimisation. The construction of the surrogate trajectories is described first. Their displacement structure and the resulting supervision are then analysed. 5.1

Bézier Surrogate Trajectories

Surrogate trajectories are constructed between model initialisations and their corresponding trained solutions  (m) (m) M using quadratic Bézier curves. Let (θ0 , θT ) m=1 denote initialisation–solution pairs obtained from real-data SGD trajectories. For each pair, define a quadratic Bézier trajectory (m)

Φ(m) (t) = (1 − t)2 θ0

(m)

+ 2t(1 − t)ϕ(m) + t2 θT ,

t ∈ [0, 1],

(10)

where ϕ(m) ∈ Rp is a learnable control point. The straight line is recovered by the special choice (m) (m)  ϕ(m) = 12 θ0 + θT , so the quadratic family strictly contains the linear interpolation baseline. Each control point is optimised to reduce the average loss along the path, Z 1  (m) L Φϕ (t); D dt, ϕ(m)⋆ ∈ arg min ϕ

(11)

0

where L(θ; D) denotes the dataset-level training loss evaluated on D. In practice, the integral is approximated using discrete samples of t ∈ [0, 1]. This optimisation is related to mode connectivity (Garipov et al., 2018), but differs in an important respect. Because one endpoint is the random initialisation, the curve is not interpreted as a classical mode connection between two low-loss modes. 7

Instead, minimising the average loss along the path encourages a smoother curve with lower average loss and, empirically, a typically decreasing expected loss profile from initialisation to solution. (m)

(m)

Each surrogate trajectory is specified by three parameter vectors, θ0 , ϕ(m)⋆ , and θT . Storing M surrogates therefore requires O(3M ) parameter vectors. Standard trajectory matching, by contrast, stores T checkpoints per trajectory and therefore requires O(T M ) parameter vectors. The surrogate representation thus yields a substantial reduction in memory when T ≫ 1. 5.2

Low-Rank Supervision from Bézier Surrogates

The surrogate trajectories above induce the displacement supervision used in trajectory matching. For a surrogate trajectory Φ(m) , define segment displacements ∆(m) (ts , te ) = Φ(m) (te ) − Φ(m) (ts ),

0 ≤ ts < te ≤ 1.

(12)

Collecting such displacements across sampled segments yields a surrogate displacement matrix analogous to A in Section 4. The key structural property is that all segment displacements from a single quadratic Bézier trajectory lie in a two-dimensional subspace. For notational simplicity, the surrogate index m is suppressed in the theorem statement. Theorem 3 (Low-dimensional displacement structure of quadratic Bézier surrogates). Let Φ(t) be a quadratic Bézier trajectory as defined in equation 10. Then for any 0 ≤ ts < te ≤ 1,  Φ(te ) − Φ(ts ) = (te − ts )(θT − θ0 ) + (te − ts )(1 − ts − te ) 2ϕ − θ0 − θT . (13) Consequently, every segment displacement lies in the subspace n o span θT − θ0 , 2ϕ − θ0 − θT , and therefore  VΦ := span Φ(te ) − Φ(ts ) : 0 ≤ ts < te ≤ 1 satisfies dim(VΦ ) ≤ 2. In particular, any displacement matrix constructed from segments of a single surrogate trajectory has rank at most 2. Proof. Section B.3 of the appendix provides the complete proof. For M surrogate trajectories, the combined supervision therefore has nominal rank at most 2M . Empirically, however, the effective rank is substantially lower and typically scales closer to M than to 2M (Figure 2). BTM therefore replaces diffuse high-rank supervision with a controlled low-rank alternative that is better aligned with the constraints of student optimisation. Rather than matching a broad spectrum of weakly aligned directions, the synthetic dataset needs only to reproduce a small number of coherent displacement directions that capture the dominant structure of optimisation. 5.3

Spectral Structure of Trajectory Supervision

The geometric analysis in Section 4 shows that the difficulty of trajectory matching depends not only on the magnitude of teacher displacements, but also on how their spectral energy is distributed. By Theorem 2, for any fixed effective reachable dimension r, slower spectral concentration of the displacement matrix A implies larger tail energy beyond its first r singular directions and therefore a stronger conditional representability bottleneck. The low-dimensional displacement structure established above suggests that Bézier surrogates should induce supervision that is more spectrally concentrated than raw SGD trajectories. Figure 2 compares the cumulative spectral energy of displacement matrices computed from 50 teacher trajectories per method across datasets. Raw SGD supervision exhibits slow spectral concentration, with energy spread across many weakly aligned directions. Convexified trajectories are more concentrated but remain relatively diffuse, whereas Bézier surrogates concentrate most of their spectral mass in far fewer 8

668

0.8 0.6 0.4

SGD Convexified Bézier

0.2 0.0

200

400

600

Number of Singular Directions

71 138 1.0

2097

0.8 0.6 0.4

SGD Convexified Bézier

0.2 0.0

500

1000

1500

2000

Number of Singular Directions

(a) Portsmouth (NHS)

(b) eICU

Cumulative Spectral Energy

63 119

Cumulative Spectral Energy

Cumulative Spectral Energy

1.0

86172 1.0

2745

0.8 0.6 0.4 0.2 0.0

SGD Convexified Bézier 500 1000 1500 2000 2500 3000

Number of Singular Directions (c) MIMIC-III

Figure 2: Effective dimensionality of teacher displacement supervision across datasets. Each panel shows the cumulative spectral energy of the displacement matrix as a function of the number of singular directions, computed from 50 teacher trajectories per method. SGD supervision exhibits high effective rank, with energy spread across many directions. In contrast, Bézier surrogates concentrate spectral mass in a few directions, while convexified trajectories are intermediate. directions. By Theorem 2, faster spectral concentration means that a larger fraction of the supervision signal can be captured within any fixed reachable span of dimension r, leaving a smaller residual tail and therefore a weaker conditional representability bottleneck. This comparison should therefore be interpreted as a teacher-side diagnostic of the bottleneck, since it characterises the spectrum of A in practice rather than the student’s reachable span itself. This contrast in spectral concentration is especially relevant in clinical settings, where low-prevalence outcomes often give rise to sparse and inconsistently aligned minority-class updates. These can introduce low-energy but task-relevant components into the supervision signal, making faithful reproduction more difficult for a compact synthetic dataset. The stronger concentration observed for Bézier surrogates is therefore consistent with the design goal of BTM, namely to preserve dominant, functionally meaningful progress while suppressing diffuse high-rank variability that is harder for a fixed synthetic dataset to reproduce. 5.4

Bézier Trajectory Matching Objective

BTM replaces the teacher displacements ∆k in trajectory matching with segment displacements sampled from learned Bézier surrogates, as defined in equation 12. In all other respects, the training pipeline follows standard trajectory matching. Assume access to a collection of M optimised Bézier surrogates {Φ(m) }M m=1 constructed from real-data |D̃| training, together with a synthetic dataset D̃ = {(x̃i , ỹi )}i=1 . The synthetic inputs are stacked into a matrix X̃ ∈ R|D̃|×dx , and the corresponding labels into a tensor Ỹ ∈ R|D̃|×C . At each outer iteration, a surrogate index m ∼ U{1, . . . , M } and time parameters 0 ≤ ts < te ≤ 1 are sampled, and the corresponding segment endpoints are set to θs = Φ(m) (ts ),

θe = Φ(m) (te ).

(14)

A student model is initialised at θ̂0 = θs and trained for N steps using the update in equation 2, producing θ̂N . In practice, optimisation uses the normalised trajectory matching loss LBTM =

∥θ̂N − θe ∥22 . ∥θs − θe ∥22

(15)

For a fixed sampled segment, the denominator depends only on the surrogate endpoints and is therefore a positive constant with respect to the student trajectory. Consequently, the student displacement that minimises equation 15 is the same as the one that minimises the unnormalised residual in equation 3, so the representability analysis in Section 4 still applies at the segment level. The effect of the normalisation is to 9

reduce scale variation across segments, making short and long segments more comparable, while reweighting them by inverse squared segment length. It should therefore not be interpreted as a purely directional objective, but rather as a stabilised implementation of the same underlying segment-level matching problem. The synthetic dataset is optimised by differentiating LBTM through the unrolled student updates. Following standard first-order trajectory matching, the meta-gradient is approximated using the gradient with respect to the final student parameters, 2(θ̂N − θe ) gL = ∇θ̂N LBTM = . (16) ∥θs − θe ∥22 The corresponding approximate meta-gradient with respect to the synthetic inputs is ∇X̃ LBTM ≈ −ηs

N −1 X

1 |B n| n=0

X

D E ∇X̃ ∇θ ℓ(θ̂n ; x̃, ỹ), gL ,

(17)

(x̃,ỹ)∈Bn

where ηs is the student learning rate and Bn denotes the synthetic mini-batch at step n. The synthetic inputs are updated by gradient descent, X̃ ← X̃ − ηx ∇X̃ LBTM ,

(18)

and the labels Ỹ can also be optimised analogously. The complete procedure is given in Algorithm 1 in the appendix. 5.5

Functional Justification of Surrogate Supervision

The low-rank analysis above concerns the geometry of the supervision signal. A complementary question is whether the surrogate path remains functionally close to the teacher supervision it replaces. The next result gives a chord-based bound showing that a quadratic Bézier surrogate has controlled curvature and remains close in prediction space to the piecewise-linear interpolation of the teacher checkpoints whenever the surrogate stays near the endpoint chord and the model is parameter-Lipschitz along the relevant paths. Theorem 4 (Geometric regularity and prediction fidelity of Bézier surrogates). Let {θτ }Tτ=0 ⊂ Rp denote a teacher trajectory with initialisation θ0 and endpoint θT , and let γ : [0, 1] → Rp denote its piecewise-linear interpolation. Define the endpoint chord c(t) := (1 − t)θ0 + tθT , and let Φ(t) := Φϕ⋆ (t) denote the optimised quadratic Bézier surrogate. Define κ := 2∥θ0 − 2ϕ⋆ + θT ∥2 ,

D := sup ∥γ(t) − c(t)∥2 . t∈[0,1]

Assume that for every x ∈ X , the model map fθ (x) is Lf -Lipschitz in θ on the set {Φ(t), γ(t) : t ∈ [0, 1]}. Then the following hold. (i) Smooth curvature. Φ′′ (t) = 2(θ0 − 2ϕ⋆ + θT ),

sup ∥Φ′′ (t)∥2 = κ.

(19)

t∈[0,1]

Moreover, sup ∥Φ(t) − c(t)∥2 = t∈[0,1]

κ . 8

(20)

(ii) Prediction fidelity. sup

∥fΦ(t) (x) − fγ(t) (x)∥2 ≤ Lf

x∈X , t∈[0,1]

10

κ 8

 +D .

(21)

Proof. Sections B.4 and B.5 of the appendix provides the complete proof.

The theorem compares the surrogate Φ with the piecewise-linear teacher interpolation γ. The chord c serves only as a geometric reference used to control this deviation. The bound is therefore informative when both the surrogate curvature κ and the teacher deviation D are small. For intuition, let the teacher updates satisfy θτ +1 = θτ − ηuτ , where η > 0 is a step size and uτ is the update direction at step τ . Define the discrete second-order difference hτ := θτ +1 − 2θτ + θτ −1 . Then hτ = −η(uτ − uτ −1 ), so local curvature of the teacher path is governed by step-to-step changes in the update direction. If uτ = dτ + ξτ , where dτ denotes systematic drift and ξτ denotes stochastic fluctuation, then   hτ = −η (dτ − dτ −1 ) + (ξτ − ξτ −1 ) . Thus, discrete teacher curvature mixes optimisation drift with stochastic variability, whereas the quadratic surrogate has globally smooth curvature. Corollary 1 (Conditional supervision sufficiency). Assume the setting of Theorem 4, and further assume that for every x ∈ X , the model map fθ (x) is Lf -Lipschitz in θ on the set {γstu (t), Φ(t), γ(t) : t ∈ [0, 1]}. Let γstu : [0, 1] → Rp denote a student trajectory induced by the synthetic dataset, and suppose that sup ∥γstu (t) − Φ(t)∥2 ≤ εsyn . t∈[0,1]

Then   κ ∥fγstu (t) (x) − fγ(t) (x)∥2 ≤ Lf εsyn + + D . 8 x∈X , t∈[0,1] sup

(22)

Proof. Section B.6 of the appendix provides the complete proof.

Corollary 1 is a conditional sufficiency statement rather than an optimisation guarantee for Algorithm 1. It identifies a condition under which a student trajectory that tracks the surrogate also remains close to the teacher interpolation in prediction space. Together, Theorem 4 and Corollary 1 show that the surrogate family is not only low-rank, but also functionally meaningful under explicit geometric conditions. Along with the empirical ablations in Section 7.4, this supports BTM as a low-complexity surrogate family that preserves endpoint-defined structure while filtering stepwise variability from teacher supervision.

6

Experimental Setup

Datasets. Evaluation is conducted on five real-world clinical datasets spanning multiple institutions, modalities, and prediction settings. These include three large de-identified emergency department cohorts collected from UK NHS Trusts in Oxford, Portsmouth, and Birmingham (Soltan et al., 2024), together with two widely used public ICU benchmarks, eICU (Pollard et al., 2018) and MIMIC-III (Johnson et al., 2016). As summarised in Table 1, the resulting benchmark suite covers both tabular and multivariate time-series data, and includes binary and multi-label classification tasks. This combination enables evaluation across both institution-specific operational datasets and established public critical-care benchmarks, providing a diverse and clinically realistic test bed for dataset condensation. Additional dataset details are provided in Section C. 11

Table 1: Summary of datasets used in this work. Source

Dataset

Modality

# Samples

UK NHS ED Cohorts

Oxford Portsmouth Birmingham

Tabular Tabular Tabular

161,955 38,717 95,236

COVID-19 diagnosis

eICU

Tabular

49,305

In-hospital mortality

Time-series

21,156 21,728

In-hospital mortality Phenotyping (25 classes, multi-label)

Public ICU Benchmarks

MIMIC-III

Task

Clinical datasets differ from standard vision benchmarks in ways that are directly relevant to trajectory matching. Outcomes are often highly imbalanced, and patient populations are heterogeneous and noisy, leading to optimisation dynamics with sparse and inconsistently aligned gradient signals. As discussed in Section 4.3, this tends to produce diffuse supervision spread across many weakly reinforced directions, which is difficult to capture within the limited reachable span of a compact synthetic dataset. Clinical settings therefore provide a particularly stringent and realistic test bed for trajectory-based condensation. Baselines. The comparison focuses on trajectory-matching-based dataset condensation methods, as these provide the most direct point of reference for BTM. The selected methods differ primarily in how the trajectory supervision signal is constructed or modified. The baselines include Matching Training Trajectories (MTT) (Cazenavette et al., 2022), which directly matches multi-step parameter updates between real and synthetic training; Flat Trajectory Distillation (FTD) (Du et al., 2023), which promotes flatter trajectories to mitigate accumulated matching error; Difficulty-Aligned Trajectory Matching (DATM) (Guo et al., 2024), which samples trajectory segments from different training stages to vary the difficulty of the supervision signal; and Matching Convexified Trajectories (MCT) (Zhong et al., 2025), which replaces SGD trajectories with convexified surrogates to improve stability and efficiency. TrajEctory matching with Soft Label Assignment (TESLA) (Cui et al., 2023) is not included, as it primarily addresses scalability by reducing memory requirements with respect to the number of unrolled optimisation steps rather than modifying the structure of the supervision signal itself. Random subset selection is included as a lower-bound baseline, and full-dataset training as an upper bound. Experimental Protocol. Synthetic datasets are constructed at 50, 100, 200, and 500 instances per class (ipc). For multi-label phenotyping, we interpret ipc as a per-label budget (plb) and map it to the total synthetic dataset size as |D̃| = plb × Lc̄ , where L is the number of labels and c̄ is the average label cardinality (i.e., the mean number of positive labels per sample). This ensures that the effective supervision budget is comparable to the single-label setting while accounting for label overlap. All methods are adapted to clinical data modalities following their original implementations, with fixed hard labels used throughout. Across all datasets, 65%, 15%, and 20% of the samples are used for training, validation, and testing, respectively. Following standard evaluation practice, randomly initialised models are trained from scratch on the condensed data and evaluated on the held-out test sets. Performance is measured using the area under the receiver operating characteristic curve (AUROC) and the area under the precision–recall curve (AUPRC). For the multi-label phenotyping, we report macro-averaged AUROC and AUPRC across labels. All results are reported over 10 random initialisations. Teacher trajectories are obtained by training models on the real training data with standard optimisation, with hyperparameters such as the learning rate, optimiser, and training horizon selected on the validation set. The main BTM-specific hyperparameters, including the number of student optimisation steps, the segment length (te − ts ), and the optimisation settings for the Bézier control point, are selected in the same way. Full implementation details are provided in Section D of the appendix. Evaluation architectures are fixed across all methods. A shallow multi-layer perceptron (MLP) (Rumelhart et al., 1986) is used for tabular data, and a temporal convolutional network (TCN) (Bai et al., 2018) for 12

Table 2: Performance on the three NHS cohorts across different ipc levels for COVID-19 prediction. Best results at each ipc are highlighted in blue (ours) and red (baseline). (a) Portsmouth dataset (5.3% prevalence) AUROC Method Random MTT FTD MCT DATM BTM (Ours)

AUPRC

50

100

200

500

50

100

200

500

0.835±0.006 0.853±0.002 0.835±0.002 0.867±0.001 0.872±0.001 0.876±0.002

0.863±0.008 0.868±0.001 0.871±0.001 0.879±0.001 0.884±0.001 0.880±0.002

0.880±0.003 0.874±0.001 0.883±0.001 0.885±0.001 0.879±0.001 0.907±0.001

0.877±0.008 0.900±0.001 0.895±0.001 0.893±0.001 0.890±0.003 0.901±0.001

0.289±0.018 0.518±0.001 0.498±0.002 0.476±0.001 0.561±0.001 0.557±0.001

0.314±0.025 0.542±0.004 0.532±0.002 0.493±0.003 0.560±0.001 0.572±0.001

0.388±0.016 0.530±0.001 0.549±0.001 0.508±0.001 0.563±0.002 0.596±0.001

0.436±0.026 0.526±0.001 0.590±0.001 0.541±0.005 0.580±0.001 0.609±0.001

Full Dataset

0.906±0.002

0.610±0.004

(b) Oxford dataset (1.7% prevalence) AUROC Method Random MTT FTD MCT DATM BTM (Ours)

AUPRC

50

100

200

500

50

100

200

500

0.831±0.003 0.834±0.002 0.851±0.008 0.845±0.001 0.856±0.005 0.865±0.004

0.856±0.004 0.861±0.004 0.850±0.006 0.862±0.003 0.868±0.004 0.867±0.009

0.870±0.005 0.855±0.008 0.852±0.005 0.882±0.001 0.875±0.003 0.886±0.005

0.879±0.007 0.869±0.001 0.874±0.006 0.874±0.003 0.880±0.004 0.891±0.001

0.149±0.002 0.370±0.008 0.368±0.001 0.341±0.014 0.359±0.015 0.414±0.001

0.171±0.006 0.391±0.006 0.385±0.009 0.369±0.016 0.413±0.005 0.425±0.002

0.230±0.008 0.405±0.012 0.400±0.009 0.358±0.004 0.419±0.002 0.435±0.001

0.256±0.007 0.410±0.007 0.410±0.005 0.352±0.002 0.415±0.003 0.440±0.001

Full Dataset

0.901±0.001

0.445±0.004

(c) Birmingham dataset (0.8% prevalence) AUROC Method Random MTT FTD MCT DATM BTM (Ours)

AUPRC

50

100

200

500

50

100

200

500

0.842±0.011 0.828±0.006 0.829±0.005 0.860±0.001 0.824±0.001 0.863±0.004

0.857±0.014 0.847±0.013 0.839±0.010 0.862±0.007 0.832±0.022 0.872±0.004

0.865±0.009 0.851±0.013 0.847±0.010 0.877±0.012 0.851±0.016 0.883±0.004

0.890±0.12 0.884±0.007 0.891±0.004 0.886±0.001 0.870±0.005 0.890±0.003

0.080±0.018 0.189±0.027 0.193±0.008 0.233±0.002 0.178±0.004 0.269±0.007

0.073±0.009 0.192±0.021 0.218±0.013 0.220±0.023 0.229±0.005 0.270±0.002

0.119±0.016 0.212±0.016 0.216±0.018 0.234±0.003 0.235±0.002 0.274±0.003

0.151±0.009 0.234±0.017 0.229±0.014 0.247±0.006 0.240±0.006 0.284±0.002

Full Dataset

0.898±0.002

0.293±0.007

time-series data. Consistent with prior work, the same architecture is used for both condensation and evaluation, except in cross-architecture experiments. Synthetic inputs are initialised from real samples.

7

Results and Discussion

7.1

Downstream Utility of Condensed Data

The primary evaluation criterion is downstream task performance after training on condensed data. Particular emphasis is placed on low-data and low-prevalence regimes, where diffuse supervision and weakly aligned gradients make trajectory-based condensation especially challenging. 7.1.1

NHS emergency department cohorts.

Table 2 reports results on the three NHS cohorts across synthetic budgets ranging from 50 to 500 instances per class for the task of COVID-19 prediction. BTM performs strongly across all settings and is particularly effective in the clinically relevant low-prevalence regime. The gains are most consistent in AUPRC, suggesting that the surrogate supervision is especially beneficial when positive-class signal is rare and difficult to preserve under severe compression. On the Portsmouth dataset (5.3% prevalence), BTM achieves the best AUROC at ipc=50, 200, and 500, and the best AUPRC from ipc=100 onward. In AUPRC, this corresponds to improvements of 2.1%, 5.9%, and 3.2% over the strongest baseline at ipc=100, 200, and 500, respectively. At ipc=500, BTM reaches 0.609, 13

Table 3: Performance on the eICU and MIMIC-III datasets across different ipc levels for in-hospital mortality prediction. Best results at each ipc are highlighted in blue (ours) and red (baseline). (a) eICU dataset AUROC Method Random MTT FTD MCT DATM BTM (Ours)

AUPRC

50

100

200

500

50

100

200

500

0.740±0.022 0.754±0.003 0.798±0.002 0.773±0.004 0.854±0.001 0.854±0.002

0.763±0.013 0.800±0.001 0.803±0.002 0.828±0.001 0.852±0.001 0.859±0.001

0.804±0.008 0.829±0.004 0.824±0.001 0.837±0.002 0.846±0.001 0.861±0.001

0.840±0.005 0.849±0.001 0.847±0.004 0.847±0.001 0.856±0.001 0.874±0.020

0.277±0.027 0.329±0.003 0.374±0.002 0.397±0.005 0.462±0.001 0.479±0.002

0.308±0.022 0.390±0.004 0.389±0.001 0.427±0.001 0.451±0.003 0.486±0.001

0.368±0.017 0.398±0.007 0.430±0.002 0.448±0.001 0.464±0.001 0.476±0.001

0.429±0.013 0.454±0.002 0.442±0.004 0.464±0.001 0.482±0.001 0.506±0.001

Full Dataset

0.879±0.002

0.515±0.003

(b) MIMIC-III dataset AUROC Method Random MTT FTD MCT DATM BTM (Ours)

AUPRC

50

100

200

500

50

100

200

500

0.740±0.022 0.800±0.008 0.770±0.003 0.821±0.007 0.824±0.003 0.828±0.005

0.763±0.013 0.821±0.003 0.779±0.005 0.831±0.002 0.834±0.002 0.835±0.003

0.804±0.008 0.834±0.002 0.811±0.005 0.835±0.002 0.838±0.002 0.839±0.002

0.840±0.005 0.838±0.002 0.828±0.003 0.840±0.002 0.842±0.002 0.840±0.002

0.307±0.022 0.421±0.021 0.361±0.009 0.453±0.013 0.461±0.009 0.471±0.008

0.338±0.022 0.449±0.013 0.384±0.012 0.481±0.009 0.484±0.008 0.488±0.007

0.368±0.017 0.493±0.006 0.443±0.011 0.492±0.007 0.500±0.007 0.499±0.006

0.409±0.013 0.492±0.007 0.473±0.009 0.493±0.005 0.496±0.006 0.503±0.003

Full Dataset

0.837±0.003

0.499±0.008

which is 99.8% of full-dataset performance (0.610). At the smallest budget, DATM is marginally stronger in AUPRC, but BTM overtakes it as the synthetic budget increases. On the Oxford dataset (1.7% prevalence), BTM again achieves the best AUROC at ipc=50, 200, and 500, and attains the strongest AUPRC at every budget. The relative AUPRC gains over the strongest baseline are 11.9%, 2.9%, 3.8%, and 6.0% at ipc=50, 100, 200, and 500, respectively. Performance improves steadily from 0.414 at ipc=50 to 0.440 at ipc=500, corresponding to 98.9% of the full-dataset AUPRC (0.445). The clearest gains appear on the Birmingham dataset (0.8% prevalence), the most imbalanced cohort. Here, BTM delivers the best AUPRC at every budget and the best AUROC up to ipc=200, while remaining competitive at ipc=500. The AUPRC gains over the strongest baseline are 15.5%, 17.9%, 16.6%, and 15.0% across the four budgets, highlighting the benefit of BTM in the most data-sparse and low-prevalence setting. At ipc=500, BTM reaches 0.284, or 96.9% of full-dataset AUPRC (0.293). This pattern is consistent with the geometric analysis in Section 4. As prevalence decreases, the supervision signal becomes increasingly diffuse and weakly aligned, making trajectory matching more sensitive to high-rank variability. BTM mitigates this effect by replacing raw SGD supervision with smoother, low-rank surrogate trajectories that retain task-relevant structure while suppressing stepwise noise. Among the baselines, MCT is generally the most competitive in AUROC, suggesting that reducing supervision rank is already beneficial. Its gains in AUPRC, however, are less consistent, particularly on the lowerprevalence cohorts. This aligns with the methodological difference between the two approaches. MCT constructs convexified trajectory surrogates from stored checkpoints, whereas BTM learns Bézier control points that reduce average path loss and encourage a smoother descent profile. DATM is also competitive on the Portsmouth and Oxford datasets, especially at smaller budgets, but its advantage weakens as prevalence decreases. Overall, these results indicate that learned low-rank surrogate supervision is particularly useful in the low-data, low-prevalence regimes most relevant to clinical practice. 7.1.2

Public ICU benchmarks.

Table 3 reports results on the eICU and MIMIC-III datasets across synthetic budgets ranging from 50 to 500 instances per class for the task of in-hospital mortality prediction. As with the NHS cohorts, BTM performs 14

Table 4: Performance on the MIMIC-III dataset across different ipc levels for 25-class multi-label phenotyping. Best results at each ipc are highlighted in blue (ours) and red (baseline). MIMIC-III dataset (Phenotyping) Macro AUROC Method Random MTT FTD MCT DATM BTM (Ours) Full Dataset

Macro AUPRC

50

100

200

500

50

100

200

500

0.602±0.005 0.668±0.004 0.659±0.005 0.675±0.004 0.679±0.004 0.690±0.003

0.640±0.005 0.689±0.004 0.681±0.004 0.698±0.004 0.700±0.003 0.713±0.003

0.661±0.001 0.705±0.003 0.699±0.004 0.713±0.003 0.715±0.003 0.727±0.002

0.687±0.002 0.717±0.003 0.714±0.003 0.714±0.002 0.730±0.002 0.729±0.002

0.261±0.005 0.310±0.008 0.299±0.008 0.321±0.007 0.327±0.007 0.339±0.006

0.290±0.004 0.336±0.006 0.325±0.007 0.342±0.006 0.352±0.006 0.365±0.005

0.312±0.002 0.358±0.005 0.341±0.006 0.361±0.005 0.371±0.004 0.383±0.004

0.340±0.002 0.372±0.004 0.366±0.005 0.369±0.004 0.379±0.003 0.389±0.003

0.738±0.001

0.395±0.002

strongly across compression levels, with the clearest and most consistent gains appearing in AUPRC. This again suggests that structured low-rank surrogate supervision is particularly effective when the clinically relevant signal is sparse and must be retained under aggressive data compression. On the eICU dataset, BTM attains the best or tied-best AUROC at every budget and the strongest AUPRC throughout. In AUPRC, the relative gains over the strongest baseline are 3.7%, 7.8%, 2.6%, and 5.0% at ipc=50, 100, 200, and 500, respectively. At the largest budget, BTM reaches 0.506, corresponding to 98.3% of full-dataset AUPRC (0.515). These results indicate that the learned Bézier surrogates retain task-relevant supervision particularly well even at low synthetic budgets. On the MIMIC-III dataset, BTM achieves the best AUROC at ipc=50, 100, and 200, while remaining competitive at ipc=500. In AUPRC, BTM is best at ipc=50, 100, and 500, with relative gains of 2.2%, 0.8%, and 1.4% over the strongest baseline at those budgets. At ipc=200, DATM is marginally stronger by 0.001 AUPRC, indicating that both methods are highly competitive in this regime. At ipc=500, BTM reaches 0.503, slightly exceeding the full-dataset AUPRC of 0.499 within experimental variability. Among the baselines, DATM and MCT are generally the strongest competitors on these public ICU benchmarks. However, BTM remains the most consistent overall in AUPRC, particularly on eICU, and is either best or near-best on MIMIC-III across all budgets. FTD appears less effective on MIMIC-III, which may reflect the structured time-series setting with a TCN, where encouraging uniformly flat trajectories can suppress temporally localised signals. Overall, the eICU and MIMIC-III results reinforce the same conclusion as the NHS cohorts that replacing raw SGD supervision with learned low-rank surrogate trajectories improves the utility of condensed datasets, especially in low-data settings. MIMIC-III phenotyping. Table 4 reports results on the MIMIC-III dataset for 25-class multi-label phenotyping across synthetic budgets ranging from 50 to 500 instances per class, which we map to a per-label budget as described in Section 6. As in the binary prediction tasks, BTM performs strongly across all compression levels, with the clearest and most consistent gains appearing in macro AUPRC. This suggests that structured surrogate supervision is effective at preserving label-specific predictive signal under severe compression in the multi-label setting. In terms of macro AUROC, BTM achieves the best performance at ipc=50, 100, and 200, with values of 0.690, 0.713, and 0.727, respectively. At ipc=500, DATM is marginally stronger (0.730 versus 0.729), indicating that the two methods are highly competitive at larger synthetic budgets. Overall, the AUROC gains are smaller than in the binary tasks, which is consistent with the greater difficulty of matching supervision in the multi-label setting. The improvements are more pronounced in macro AUPRC. BTM attains the best performance at every budget, exceeding the strongest baseline by 3.7%, 3.7%, 3.2%, and 2.7% at ipc=50, 100, 200, and 500, respectively. Macro AUPRC increases steadily from 0.339 at ipc=50 to 0.389 at ipc=500, reaching 98.5% of full-dataset performance (0.395). This pattern suggests that BTM is particularly effective at retaining weaker or less consistently reinforced label signals, which are especially important for precision–recall performance in multi-label clinical prediction. 15

(a) eICU

(b) MIMIC-III

0.510

DATM BTM

0.505

0.48

0.495

AUPRC

AUPRC

0.500 0.490 0.485

0.46 0.44

0.480 0.475 0.470

DATM BTM

0.50

0.42 MLP MLP-1 (condensation)

MLP-2

MLP-3

MLP-4

TCN TCN-1 (condensation)

TCN-2

LSTM-1

LSTM-2

Figure 3: Cross-architecture generalisation at ipc=500. Synthetic datasets are condensed using a single source architecture for each dataset (shaded) and then evaluated on unseen target architectures. DATM is shown as the comparison baseline because it was the strongest-performing baseline on these datasets for in-hospital mortality prediction in the main experiments. BTM consistently outperforms DATM, especially under larger architecture shifts.

Among the baselines, DATM is the strongest overall competitor. It remains consistently competitive across budgets and achieves the best macro AUROC at ipc=500. MCT is also competitive in macro AUROC, but its macro AUPRC is weaker, while MTT remains below DATM and BTM on both metrics. FTD is again less effective, particularly in macro AUPRC, suggesting that enforcing flatter surrogate trajectories alone is insufficient when supervision must preserve diverse, label-specific structure. Overall, these results reinforce the same conclusion as in the binary tasks that replacing raw SGD supervision with structured, low-rank surrogate trajectories improves the utility of condensed datasets. In the multi-label setting, where supervision is inherently more diffuse because of label overlap, these benefits are most clearly reflected in macro AUPRC.

7.2

Cross-Architecture Generalisation

Cross-architecture evaluation tests whether the utility of a condensed dataset depends strongly on the architecture used during condensation. Figure 3 reports AUPRC at ipc=500. For each dataset, synthetic data are first condensed using a single source architecture, highlighted by the shaded region, and then evaluated on a set of unseen target architectures. We compare BTM only against DATM here because DATM was the strongest-performing baseline on these datasets in the main experiments, making it the most informative reference point for cross-architecture transfer. On the eICU dataset, synthetic data condensed with an MLP transfer robustly across MLP variants of different depths and widths. BTM outperforms DATM on every target architecture, with absolute AUPRC gains ranging from 0.008 to 0.024. Its performance also remains stable across the MLP family, varying only from 0.488 to 0.506. This suggests that the supervision induced by BTM is not tightly coupled to a specific parametrisation of the MLP, but instead captures task-relevant signal that generalises across closely related architectures. The transfer setting is more demanding on MIMIC-III, where condensation is performed with a TCN and evaluation includes both TCN and LSTM targets. Here, architecture shift changes the temporal inductive bias of the model, making transfer substantially harder. Even in this setting, BTM remains consistently stronger than DATM across all target architectures. The gains are modest within the TCN family, but larger under transfer to LSTMs, where BTM improves AUPRC by 0.012 and 0.028 on LSTM-1 and LSTM-2, 16

Bézier Surrogates

Oxford (NHS)

SGD Trajectories

41 MB 1340 MB 59 MB

eICU

1950 MB 87 MB

MIMIC-III

1720 MB 50

100

Storage (MB, log scale)

500

1000

2000

Figure 4: Trajectory storage across clinical datasets. Compared with full SGD trajectories, Bézier surrogates substantially reduce storage requirements, yielding approximately 33× lower storage on Oxford (NHS) and eICU, and 20× lower storage on MIMIC-III.

respectively. Relative to its source-architecture performance, BTM also exhibits slightly smaller degradation than DATM under TCN-to-LSTM transfer. Overall, these results suggest that structuring the supervision signal through learned low-rank surrogate trajectories improves robustness to architecture shift. This effect is most evident when transferring across model classes with different inductive biases, where preserving task-relevant structure without overfitting to architecture-specific optimisation dynamics becomes especially important. 7.3

Storage Efficiency

Figure 4 shows that BTM substantially reduces trajectory storage across all clinical datasets. Relative to full SGD trajectories, Bézier surrogates yield approximately 33× lower storage on Oxford (NHS) and eICU, and 20× lower storage on MIMIC-III. These reductions lower memory requirements and make it easier to use more expert trajectories in resource-constrained clinical settings. 7.4

Surrogate Path Complexity Ablation

Figure 5 evaluates how surrogate trajectory parameterisation affects condensation quality. We compare three increasingly structured path families: simple linear interpolation, a convexified linear trajectory that preserves temporal structure from SGD through fixed convex weights, and the proposed quadratic Bézier curve with a learnable control point. Results are reported in AUPRC across five clinical datasets at both low (ipc=50) and high (ipc=500) synthetic budgets. A consistent trend emerges across datasets and budgets. Simple linear interpolation provides a strong baseline, indicating that reducing trajectory complexity is beneficial relative to storing and following full stochastic optimisation paths. Introducing a learnable control point yields further gains in almost every case: Bézier trajectories achieve the highest AUPRC in 9 of 10 dataset–budget settings and remain within 0.001 of the best result in the remaining case. Improvements are especially pronounced on the Oxford, Portsmouth, eICU, and MIMIC-III datasets, and persist across both low and high budgets. By contrast, the convexified trajectory often underperforms simple linear interpolation, despite preserving more of the temporal structure of SGD. This suggests that retaining trajectory structure alone is insufficient; the path must be explicitly optimised to provide a compact and informative supervision signal. Figure 6 provides a functional view of this effect by examining the loss landscape along each trajectory. While linear interpolation smooths the high-variance behaviour of SGD, it can still traverse higher-loss regions between endpoints. In contrast, the Bézier trajectory is optimised to minimise average loss along the path and remains consistently in lower-loss regions, yielding a supervision signal that is both structured and task-aligned rather than merely smoothed. 17

IPC = 50 0.6 0.5

AUPRC

IPC = 500 Linear Conv Linear Bézier

Linear Conv Linear Bézier

0.4 0.3 OUH

PUH

UHB

Dataset

eICU

MIMIC-III

OUH

PUH

UHB

eICU

Dataset

MIMIC-III

0.7 0.6 0.5 0.4 0.3 0.2 0.1 0.0

20

0.2

Epochs

40

0.4

60

0.6

Interpolation t

80 100 SGD Linear Bézier

0.8

(a) Portsmouth (NHS)

1.0

0.7

0

20

Epochs

40

60

0.6 0.5

80 100 SGD Linear Bézier

0.4 0.3 0.2 0.0

0.2

0.4

0.6

Interpolation t (b) eICU

0.8

1.0

Average Training Loss

Average Training Loss

0

Average Training Loss

Figure 5: Surrogate path complexity ablation. AUPRC across five clinical datasets at ipc=50 and ipc=500 for three surrogate trajectory parameterisations: linear interpolation, convexified linear interpolation, and quadratic Bézier curves. OUH, PUH, and UHB denote Oxford, Portsmouth, and Birmingham NHS cohorts, respectively. Error bars denote standard deviation across runs. Bézier trajectories achieve the strongest overall performance, outperforming the linear variants in 9 of 10 dataset–budget settings. 0

12

0.4 0.0

0.2

0.7

Epochs

24

36

48 60 SGD Linear Bézier

0.4

0.6

0.8

0.6 0.5

Interpolation t

1.0

(c) MIMIC-III (Phenotyping)

Figure 6: Training loss profiles along surrogate trajectories. Average training loss for linear and quadratic Bézier trajectories as a function of interpolation parameter t, and for SGD as a function of training epochs, across datasets. While linear interpolation provides a smoother and more structured path than SGD, it can still traverse higher-loss regions. In contrast, the Bézier trajectory remains consistently in lower-loss regions due to its task-optimised construction. Together, these results support the central design choice of BTM: a low-complexity surrogate path is most effective when it is both structured and task-optimised, yielding a stronger and more informative supervision signal for dataset condensation.

8

Conclusion

We presented a geometric view of trajectory matching showing that, for a fixed synthetic dataset, the trajectory-matching residual is lower bounded by the component of the supervision signal lying outside the student’s reachable gradient span. This yields a conditional rank-based representability bottleneck, where, when the supervision signal is not representable within that span, optimisation alone cannot eliminate the mismatch. This perspective clarifies a core limitation of existing trajectory-based dataset condensation methods and motivates the design of supervision signals that are better aligned with the student’s optimisation geometry. Motivated by this insight, we introduced Bézier Trajectory Matching (BTM), which replaces long stochastic optimisation trajectories with structured, low-rank surrogates. By learning quadratic Bézier paths that reduce average path loss and encourage a smooth descent profile from initialisation to solution, BTM provides a more structured supervision signal while preserving functionally meaningful training dynamics. Across 18

diverse clinical datasets spanning tabular and time-series modalities, BTM consistently improves predictive performance, with the strongest gains in low-prevalence and low-data regimes. It also remains robust under architecture shift and reduces trajectory storage requirements by up to 33×, improving the practicality of trajectory-based condensation in resource-constrained clinical settings. More broadly, our results suggest that effective dataset condensation depends not only on how much supervision is provided, but also on how that supervision is structured. In clinical contexts, this is particularly important because compact synthetic datasets may help support more efficient reuse of information derived from governed datasets while reducing storage and training costs. BTM currently uses a fixed quadratic parameterisation, and extending it to richer adaptive trajectory families is a natural next step. Incorporating differential privacy is another important direction for enabling safe and scalable clinical deployment. Broader Impact Statement This work has potential positive impact in machine learning settings where access to large real-world datasets is constrained by governance, storage, or computational cost. By improving dataset condensation, BTM may reduce the resources required to store, transfer, and repeatedly train on large datasets, thereby supporting more reproducible experimentation, lowering barriers for smaller research groups, and enabling more sustainable model development. In clinical machine learning, these benefits are especially relevant because access to governed health data is often restricted, so compact synthetic datasets may help support broader methodological research on information derived from such data in resource-constrained academic and clinical environments. These benefits should be balanced against important risks. Condensed or synthetic datasets are not inherently safe to share, and BTM does not provide a formal privacy guarantee or replace privacy-preserving methods, data governance procedures, or access controls. Such datasets may also retain or amplify demographic imbalance, site-specific artefacts, and historical bias, which could lead to misleading conclusions or uneven downstream performance across patient subgroups. More capable condensation methods could also be misused to create compact surrogates of sensitive datasets that are redistributed or deployed without adequate scrutiny of privacy, fairness, or clinical validity. We therefore view BTM as a method for improving the utility of trajectory-based dataset condensation, rather than as a standalone solution for safe data sharing or clinical deployment. Any practical use should be accompanied by explicit privacy assessment, subgroup-level fairness evaluation, and task-specific external validation. An important direction for future work is to combine structured trajectory-based condensation with formal privacy guarantees and stronger safeguards against bias, misuse, and over-interpretation of condensed data.

References Shaojie Bai, J Zico Kolter, and Vladlen Koltun. An empirical evaluation of generic convolutional and recurrent networks for sequence modeling. arXiv preprint arXiv:1803.01271, 2018. George Cazenavette, Tongzhou Wang, Antonio Torralba, Alexei A. Efros, and Jun-Yan Zhu. Dataset distillation by matching training trajectories. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp. 10718–10727, 2022. Justin Cui, Ruochen Wang, Si Si, and Cho-Jui Hsieh. Scaling up dataset distillation to imagenet-1k with constant memory. In International Conference on Machine Learning, pp. 6565–6590. PMLR, 2023. Felix Draxler, Kambis Veschgini, Manfred Salmhofer, and Fred A Hamprecht. Essentially no barriers in neural network energy landscape. In International Conference on Machine Learning, pp. 1309–1318. PMLR, 2018. Jiawei Du, Yidi Jiang, Vincent YF Tan, Joey Tianyi Zhou, and Haizhou Li. Minimizing the accumulated trajectory error to improve dataset distillation. arXiv preprint arXiv:2211.11004, 2023. Timur Garipov, Pavel Izmailov, Dmitrii Podoprikhin, Dmitry P Vetrov, and Andrew Gordon Wilson. Loss surfaces, mode connectivity, and fast ensembling of dnns. In Advances in Neural Information Processing Systems, volume 31, 2018. 19

Ziyao Guo, Kai Wang, George Cazenavette, Hui Li, Kaipeng Zhang, and Yang You. Towards lossless dataset distillation via difficulty-aligned trajectory matching. In International Conference on Learning Representations, 2024. Pavel Izmailov, Dmitrii Podoprikhin, Timur Garipov, Dmitry Vetrov, and Andrew Gordon Wilson. Averaging weights leads to wider optima and better generalization. In Conference on Uncertainty in Artificial Intelligence, pp. 876–885. PMLR, 2018. A. E. W. Johnson, T. J. Pollard, L. Shen, L. H. Li-wei, M. Feng, M. Ghassemi, B. Moody, P. Szolovits, Anthony L. Celi, and R. G. Mark. Mimic-iii, a freely accessible critical care database. Scientific Data, 3(1): 1–9, 2016. Renuga Kanagavelu, Madhav Walia, Yuan Wang, Huazhu Fu, Qingsong Wei, Yong Liu, and Rick Siow Mong Goh. Medsynth: Leveraging generative model for healthcare data sharing. In International Conference on Medical Image Computing and Computer-Assisted Intervention, pp. 654–664. Springer, 2024. Timothy Nguyen, Zhourong Chen, and Jaehoon Lee. Dataset meta-learning from kernel ridge regression. In International Conference on Learning Representations, 2021. Timothy Nguyen, Roman Novak, Lechao Xiao, and Jaehoon Lee. Dataset distillation with infinitely wide convolutional networks. In Advances in Neural Information Processing Systems, volume 35, pp. 5186–5198, 2022. R. Pastorino, C. De Vito, G. Migliara, K. Glocker, I. Binenbaum, W. Ricciardi, and S. Boccia. Benefits and challenges of big data in healthcare: an overview of the european initiatives. European Journal of Public Health, 29(Supplement_3):23–27, 2019. T. J. Pollard, A. Johnson, J. D. Raffa, L. A. Celi, R. G. Mark, and O. Badawi. The eicu collaborative research database, a freely available multi-center database for critical care research. Scientific Data, 5(1):1–13, 2018. David E Rumelhart, Geoffrey E Hinton, and Ronald J Williams. Learning representations by back-propagating errors. Nature, 323(6088):533–536, 1986. Mahsa Shabani. The impact of the general data protection regulation (gdpr) on artificial intelligence. European Journal of Human Genetics, 27(1):15–17, 2019. Andrew AS Soltan, Anshul Thakur, Jenny Yang, Anoop Chauhan, Leon G D’Cruz, Phillip Dickson, Marina A Soltan, David R Thickett, David W Eyre, Tingting Zhu, et al. A scalable federated learning solution for secondary care using low-cost microcomputing: privacy-preserving development and evaluation of a covid-19 screening test in uk hospitals. The Lancet Digital Health, 6(2):e93–e104, 2024. Peng Sun, Bei Shi, Daiwei Yu, and Tao Lin. On the diversity and realism of distilled dataset: An efficient dataset distillation paradigm. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp. 3928–3937, 2023. Norman J. Tatro, Pin-Yu Chen, Payel Das, Igor Melnyk, Prasanna Sattigeri, and Rongjie Lai. Optimizing mode connectivity via neuron alignment. In Advances in Neural Information Processing Systems, volume 33, pp. 15300–15311, 2020. Anshul Thakur, Tingting Zhu, Vinayak Abrol, Jacob Armstrong, Yujiang Wang, and David A Clifton. Data encoding for healthcare data democratization and information leakage prevention. Nature Communications, 15(1):1582, 2024. Anshul Thakur, Soheila Molaei, Patrick Schwab, Danielle Belgrave, Kim Branson, and David A Clifton. Optimising clinical federated learning through mode connectivity-based model aggregation. In International Conference on Artificial Intelligence and Statistics, pp. 163–171. PMLR, 2025. Kai Wang, Bo Zhao, Xiangyu Peng, Zheng Zhu, Shuo Yang, Shuo Wang, Guan Huang, Hakan Bilen, Xinchao Wang, and Yang You. Cafe: Learning to condense dataset by aligning features. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp. 12196–12205, 2022. 20

Tongzhou Wang, Jun-Yan Zhu, Antonio Torralba, and Alexei A. Efros. Dataset distillation. arXiv preprint arXiv:1811.10959, 2018. Zeyuan Yin, Eric Xing, and Zhiqiang Shen. Squeeze, recover and relabel: Dataset condensation at imagenet scale from a new perspective. In Advances in Neural Information Processing Systems, volume 36, 2023. Bo Zhao and Hakan Bilen. Dataset condensation with gradient matching. In International Conference on Learning Representations, 2021a. Bo Zhao and Hakan Bilen. Dataset condensation with differentiable siamese augmentation. In Proceedings of the 38th International Conference on Machine Learning, volume 139 of Proceedings of Machine Learning Research, pp. 12674–12685. PMLR, 2021b. Bo Zhao and Hakan Bilen. Dataset condensation with distribution matching. In Proceedings of the IEEE/CVF Winter Conference on Applications of Computer Vision, pp. 6514–6523, 2023. Wenliang Zhong, Haoyu Tang, Qinghai Zheng, Mingzhu Xu, Yupeng Hu, and Weili Guan. Towards stable and storage-efficient dataset distillation: Matching convexified trajectory. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp. 25581–25589, 2025.

21

A

PIPELINE OF PROPOSED METHOD

Algorithm 1 Dataset Condensation Using Trajectory Surrogates |D̃|

(m)

Require: Collection of M optimised Bézier surrogates {Φϕ⋆ }M m=1 , synthetic dataset D̃ = {(x̃i , ỹi )}i=1 , number of classes C, input dimension d, student learning rate ηs , meta learning rate ηx , number of student steps N , batch size b, maximum iterations Tmax Ensure: Optimised synthetic dataset D̃⋆ 1: Stack synthetic inputs into matrix X̃ ∈ R|D̃|×d 2: Stack synthetic labels into tensor Ỹ ∈ R|D̃|×C 3: for i = 1 to Tmax do 4: Sample surrogate index m ∼ U{1, . . . , M } 5: Sample ts , te ∼ U(0, 1) such that ts < te (m) 6: θs ← Φϕ⋆ (ts ) {Start position on surrogate} (m)

11:

θe ← Φϕ⋆ (te ) {Target position on surrogate} θ̂0 ← θs {Initialise student model} for n = 0 to N − 1 do Sample mini-batch B n ⊂ D̃, |Bn | = b  P θ̂n+1 ← θ̂n − ηs ∇θ̂n 1b (x̃,ỹ) ℓ(θ̂n ; x̃, ỹ)

12:

end for

7: 8: 9: 10:

∈Bn

13: 14: 15:

∥θ̂N − θe ∥22 {Normalised matching loss} ∥θs − θe ∥22 2(θ̂N − θe ) gL ← ∇θ̂N LBTM = {Gradient signal} ∥θs − θe ∥22 E D P PN −1 Compute ∇X̃ LBTM ≈ −ηs n=0 |B1n | (x̃,ỹ)∈Bn ∇X̃ ∇θ ℓ(θ̂n ; x̃, ỹ), gL , LBTM ←

16: X̃ ← X̃ − ηx ∇X̃ LBTM {Update synthetic inputs} 17: end for 18: return D̃ ⋆ = D̃

B

Proofs

This section gathers the proofs of the main theoretical results. We first prove the geometric lower bounds underlying trajectory matching, namely Theorem 1 and Theorem 2, which formalise the restriction imposed by the student’s reachable span. We then prove Theorem 3, which establishes the low-dimensional structure of the displacements induced by quadratic Bézier surrogate trajectories. Finally, we prove Theorem 4 and Corollary 1. The corollary is a conditional statement rather than an algorithmic guarantee; it isolates a sufficient condition under which a synthetic trajectory would inherit the surrogate’s prediction-space behaviour. Throughout, we retain the notation introduced in the main text. All norms on parameter vectors are Euclidean unless stated otherwise, and ∥ · ∥F denotes the Frobenius norm for matrices. B.1

Proof of Theorem 1

Proof. Fix a segment k. Under the first-order, no-momentum student update θ̂n+1 = θ̂n − ηs gn ,

n = 0, . . . , N − 1,

(1)

the student displacement after N inner steps is δk = θ̂N − θsk = −ηs

N −1 X n=0

22

gn .

(2)

Since each gn belongs to the reachable gradient span Gk = span{g0 , . . . , gN −1 },

(3)

δk ∈ Gk .

(4)

it follows immediately that

By definition, the trajectory matching loss for segment k is LTM (sk , ek ; D̃) = ∥δk − ∆k ∥22 .

(5)

Thus, for fixed teacher displacement ∆k , the loss is the squared distance from ∆k to a point δk constrained to lie in Gk . Therefore LTM (sk , ek ; D̃) ≥ min ∥v − ∆k ∥22 . (6) v∈Gk

Let PGk denote the orthogonal projector onto Gk . The orthogonal decomposition of ∆k with respect to Gk is ∆k = PGk ∆k + (I − PGk )∆k ,

(7)

where (I − PGk )∆k ∈ Gk⊥ .

(8)

 v − ∆k = v − PGk ∆k − (I − PGk )∆k .

(9)

PGk ∆k ∈ Gk , Hence, for any v ∈ Gk ,

The first term lies in Gk , whereas the second lies in Gk⊥ , so the two terms are orthogonal. By the Pythagorean theorem, ∥v − ∆k ∥22 = ∥v − PGk ∆k ∥22 + ∥(I − PGk )∆k ∥22 ≥ ∥(I − PGk )∆k ∥22 . (10) Taking v = δk ∈ Gk yields LTM (sk , ek ; D̃) = ∥δk − ∆k ∥22 ≥ ∥(I − PGk )∆k ∥22 .

(11)

It remains to identify the minimiser of the best-in-subspace problem. Equality in ∥v − ∆k ∥22 = ∥v − PGk ∆k ∥22 + ∥(I − PGk )∆k ∥22

(12)

∥v − PGk ∆k ∥22 = 0,

(13)

v = PGk ∆k .

(14)

holds if and only if that is, if and only if Therefore the minimiser of minv∈Gk ∥v − ∆k ∥22 is v⋆ = PGk ∆k .

(15)

This identifies the tightest point in the reachable span. Whether the realised student displacement δk attains this minimiser depends on the synthetic-data optimisation and is not asserted here. This proves the theorem. B.2

Proof of Theorem 2

Proof. For each segment k, the student displacement satisfies δk ∈ Gk .

(16)

G = span{gn(k) : k = 1, . . . , K, n = 0, . . . , N − 1}.

(17)

Define the global reachable span by

23

Since (k)

(k)

Gk = span{g0 , . . . , gN −1 },

(18)

it follows that Gk ⊆ G for every k. Hence δk ∈ G

for all k = 1, . . . , K.

(19)

Define (k)

LTM := LTM (sk , ek ; D̃) = ∥δk − ∆k ∥22 .

(20)

Since δk ∈ G, this loss is the squared distance from ∆k to a point in G. Exactly as in the proof of Theorem 1, if PG denotes the orthogonal projector onto G, then (k)

LTM ≥ min ∥v − ∆k ∥22 = ∥(I − PG )∆k ∥22 . v∈G

(21)

Thus, for every k, (k)

LTM ≥ ∥(I − PG )∆k ∥22 .

(22)

Summing over k = 1, . . . , K gives K X

(k)

K X

∥(I − PG )∆k ∥22 .

(23)

A = [∆1 , . . . , ∆K ] ∈ Rp×K .

(24)

  (I − PG )A = (I − PG )∆1 , . . . , (I − PG )∆K .

(25)

LTM ≥

k=1

k=1

Now recall the teacher displacement matrix

Then Using the identity ∥M∥2F =

K X

∥M:,k ∥22

(26)

k=1

for any matrix M ∈ Rp×K , we obtain K X

∥(I − PG )∆k ∥22 = ∥(I − PG )A∥2F .

(27)

k=1

Therefore K X

(k)

LTM ≥ ∥(I − PG )A∥2F ,

(28)

k=1

and dividing by K yields K

1 X (k) 1 LTM ≥ ∥(I − PG )A∥2F . K K

(29)

k=1

It remains to minimise this residual over all r-dimensional subspaces G ⊂ Rp . Let the singular value decomposition of A be A = UΣV⊤ , (30) with singular values ordered as σ1 (A) ≥ σ2 (A) ≥ · · · ≥ σrA (A) > 0,

(31)

rA := rank(A).

(32)

where

24

The optimisation problem min dim(G)=r

∥(I − PG )A∥2F

(33)

is equivalent to the problem of finding the best rank-r approximation to A in Frobenius norm, since PG A has all of its columns in G and therefore has rank at most r. By the Eckart–Young–Mirsky theorem, the optimal choice is to take G to be the span of the top r left singular vectors of A, in which case the minimum residual is rA X min ∥(I − PG )A∥2F = σj (A)2 . (34) dim(G)=r

j=r+1

Dividing by K gives rA 1 1 X σj (A)2 . ∥(I − PG )A∥2F = K j=r+1 dim(G)=r K

min

(35)

This proves the theorem. B.3

Proof of Theorem 3

Proof. Consider the quadratic Bézier trajectory Φ(t) = (1 − t)2 θ0 + 2t(1 − t)ϕ + t2 θT ,

t ∈ [0, 1].

(36)

For any 0 ≤ ts < te ≤ 1, define the segment displacement ∆(ts , te ) := Φ(te ) − Φ(ts ).

(37)

We begin by rewriting Φ(t) in a form that makes its underlying linear structure explicit. Expanding the quadratic terms gives Φ(t) = θ0 + 2t(ϕ − θ0 ) + t2 (θ0 − 2ϕ + θT ). (38) Hence, for any ts , te ∈ [0, 1], ∆(ts , te ) = 2(te − ts )(ϕ − θ0 ) + (t2e − t2s )(θ0 − 2ϕ + θT ) h i = (te − ts ) 2(ϕ − θ0 ) + (te + ts )(θ0 − 2ϕ + θT ) . Therefore every segment displacement belongs to the subspace  span ϕ − θ0 , θ0 − 2ϕ + θT .

(39)

(40)

By definition,  VΦ := span Φ(te ) − Φ(ts ) : 0 ≤ ts < te ≤ 1 ,

(41)

 VΦ ⊆ span ϕ − θ0 , θ0 − 2ϕ + θT .

(42)

so we conclude that Since the latter space has dimension at most 2, it follows that dim(VΦ ) ≤ 2.

(43)

This proves the first claim. For the second claim, let (1) (K) (K) p×K AΦ = [∆(t(1) s , te ), . . . , ∆(ts , te )] ∈ R

(44)

be any displacement matrix formed from segments of Φ. Each column of AΦ lies in VΦ , and we have shown that dim(VΦ ) ≤ 2. Hence the column space of AΦ has dimension at most 2, so rank(AΦ ) ≤ 2.

(45)

Therefore any displacement matrix constructed from segments of a single quadratic Bézier trajectory has rank at most 2. 25

B.4

Auxiliary lemma for Theorem 4

We first characterise the deviation of a quadratic Bézier curve from the chord joining its endpoints. Lemma 1. Let c(t) := (1 − t)θ0 + tθT .

(46)

 Φ(t) − c(t) = t(1 − t) 2ϕ⋆ − θ0 − θT ,

(47)

Then, for all t ∈ [0, 1], and hence ∥Φ(t) − c(t)∥2 =

t(1 − t) κ. 2

(48)

κ . 8

(49)

In particular, sup ∥Φ(t) − c(t)∥2 = t∈[0,1]

Proof. Using the definitions of Φ(t) and c(t),     Φ(t) − c(t) = (1 − t)2 − (1 − t) θ0 + 2t(1 − t)ϕ⋆ + t2 − t θT .

(50)

Since (1 − t)2 − (1 − t) = −t(1 − t),

t2 − t = −t(1 − t),

(51)

we obtain  Φ(t) − c(t) = t(1 − t) 2ϕ⋆ − θ0 − θT .

(52)

Taking Euclidean norms and using the definition κ = 2∥θ0 − 2ϕ⋆ + θT ∥2 = 2∥2ϕ⋆ − θ0 − θT ∥2 ,

(53)

gives ∥Φ(t) − c(t)∥2 =

t(1 − t) κ. 2

(54)

Finally, t(1 − t) ≤ 41 for all t ∈ [0, 1], with equality at t = 12 . Therefore sup ∥Φ(t) − c(t)∥2 = t∈[0,1]

B.5

κ . 8

(55)

Proof of Theorem 4

Proof. We prove the two claims separately. (i) Smooth curvature.

Consider the optimised quadratic Bézier surrogate Φ(t) = (1 − t)2 θ0 + 2t(1 − t)ϕ⋆ + t2 θT .

(56)

Differentiating twice with respect to t yields Φ′′ (t) = 2(θ0 − 2ϕ⋆ + θT ),

(57)

sup ∥Φ′′ (t)∥2 = 2∥θ0 − 2ϕ⋆ + θT ∥2 = κ.

(58)

which is constant in t. Hence t∈[0,1]

The chord-deviation bound in Theorem 4(i) follows from Lemma 1. 26

(ii) Prediction fidelity. Fix any x ∈ X and t ∈ [0, 1]. By the Lf -Lipschitz continuity of the model map in parameter space on the set traced by Φ and γ, ∥fΦ(t) (x) − fγ(t) (x)∥2 ≤ Lf ∥Φ(t) − γ(t)∥2 .

(59)

We now bound the parameter-space discrepancy by introducing the endpoint chord c(t). By the triangle inequality, ∥Φ(t) − γ(t)∥2 ≤ ∥Φ(t) − c(t)∥2 + ∥γ(t) − c(t)∥2 . (60) By Lemma 1,

κ , 8

(61)

∥γ(t) − c(t)∥2 ≤ D.

(62)

∥Φ(t) − c(t)∥2 ≤ and by definition of D, Therefore ∥Φ(t) − γ(t)∥2 ≤

κ + D, 8

and consequently ∥fΦ(t) (x) − fγ(t) (x)∥2 ≤ Lf

κ

8 Taking the supremum over x ∈ X and t ∈ [0, 1] proves the claim. B.5.1

(63)  +D .

(64)

Teacher-path curvature intuition.

For a generic teacher update sequence of the form θτ +1 = θτ − ηuτ ,

(65)

hτ := θτ +1 − 2θτ + θτ −1 .

(66)

define the discrete second-order difference

Substituting the update at times τ and τ − 1 gives hτ = −η(uτ − uτ −1 ).

(67)

If uτ = dτ + ξτ is decomposed into systematic drift and stochastic variability, then   hτ = −η (dτ − dτ −1 ) + (ξτ − ξτ −1 ) .

(68)

Thus, discrete teacher curvature mixes optimisation drift with stochastic variability, whereas the quadratic Bézier surrogate has globally smooth curvature. B.6

Proof of Corollary 1

Proof. Fix any x ∈ X and t ∈ [0, 1]. By the triangle inequality, ∥fγstu (t) (x) − fγ(t) (x)∥2 ≤ ∥fγstu (t) (x) − fΦ(t) (x)∥2 + ∥fΦ(t) (x) − fγ(t) (x)∥2 .

(69)

Using again the Lf -Lipschitz continuity of the model map, ∥fγstu (t) (x) − fΦ(t) (x)∥2 ≤ Lf ∥γstu (t) − Φ(t)∥2 ≤ Lf εsyn .

(70)

Moreover, Theorem 4(ii) gives ∥fΦ(t) (x) − fγ(t) (x)∥2 ≤ Lf

κ 8

 +D .

(71)

Combining the two bounds yields   κ ∥fγstu (t) (x) − fγ(t) (x)∥2 ≤ Lf εsyn + + D . 8 Taking the supremum over x ∈ X and t ∈ [0, 1] proves the corollary. 27

(72)

C

DATASET DETAILS

We provide implementation-specific preprocessing and representation details for each dataset. Shared experimental settings (splits, normalisation, and evaluation protocol) are described in Section 6. C.1

NHS Emergency Department Datasets (Oxford, Portsmouth, Birmingham)

For the NHS cohorts, each patient is represented by a fixed 27-dimensional feature vector derived from admission-time data. All features correspond to measurements available at presentation, and no temporal aggregation is performed. Each dataset is treated independently, with no cross-site data mixing. Dataset-level statistics are summarised in Table 1. Each dataset comprises routinely collected admission-time features, including demographics, vital signs, and laboratory blood tests, resulting in a total of 27 features per patient. The clinical predictors are grouped by category in Table 2. Missing values are handled via median imputation. Access to the datasets is governed by NHS and Health Research Authority (HRA) approval (IRAS ID: 281832). Oxford data are available via the Infections in Oxfordshire Research Database1 , while Portsmouth and Birmingham data are accessible upon reasonable request to the respective trusts. Table 1: Dataset characteristics for NHS emergency department datasets.

# Examples # Positive Examples # Features Prevalence (%)

Oxford

Portsmouth

Birmingham

161,955 2,791 27 1.7

38,717 2,005 27 5.3

95,236 790 27 0.8

Table 2: Clinical predictors used for COVID-19 diagnosis. Category

C.2

Features

Vital Signs

Heart rate, respiratory rate, oxygen saturation, systolic blood pressure, diastolic blood pressure, temperature

Blood Tests

Haemoglobin, haematocrit, mean cell volume, white cell count, neutrophil count, lymphocyte count, monocyte count, eosinophil count, basophil count, platelets

Liver Function Tests & C-reactive protein

Albumin, alkaline phosphatase, alanine aminotransferase, bilirubin, C-reactive protein

Urea & Electrolytes

Sodium, potassium, creatinine, urea, estimated glomerular filtration rate

eICU

We use the pre-processed version of the eICU Collaborative Research Database available at https:// physionet.org/content/mimic-eicu-fiddle-feature/1.0.0/. This representation maps each ICU stay to a fixed 402-dimensional feature vector. Dataset characteristics are reported in Table 3 and variables used to construct the feature representation are summarised in Table 4. 1 https://oxfordbrc.nihr.ac.uk/research-themes/modernising-medical-microbiology-and-big-infection-diagnostics/ infections-in-oxfordshire-research-database-iord/

28

Continuous variables are discretised into quantile-based bins (typically five bins), and summary statistics (minimum, maximum, and mean) are computed where applicable. Categorical variables are one-hot encoded. This results in a fully tabular representation with no temporal dimension. Table 3: Dataset characteristics for eICU. eICU # Examples # Positive Examples # Features Prevalence (%)

49,305 4,501 402 9.1

Table 4: Source physiological and demographic variables used to construct the 402-dimensional feature vector for eICU in-hospital mortality prediction. Category

C.3

Variables

Demographics

Age, height, weight, gender

Admission Information

Hospital admit source, hospital admit offset, unit admit source, unit stay type, unit type, Apache admission diagnosis, airway type

Vital Signs

Heart rate, respiratory rate, oxygen saturation, temperature (Celsius and Fahrenheit), temperature location

Blood Pressure

Non-invasive systolic/diastolic/mean blood pressure, invasive systolic/diastolic/mean blood pressure, central venous pressure

Oxygen Support

O2 administration device, O2 level percentage

Laboratory Measurements

Glucose

MIMIC-III

We process MIMIC-III using publicly available benchmarking code to obtain multivariate time-series representations. Only patients with at least 48 hourly observations are retained, and the first 48 hours of each stay are used for all tasks. After removing duplicate features, each time step is represented by a 60-dimensional feature vector, yielding an input tensor of shape 48 × 60 per patient. Binary mask features are included to indicate the presence or absence of each measurement at each time step. Categorical clinical variables are encoded as follows: Glasgow Coma Scale components are one-hot encoded (eye opening: 5 categories; motor response: 6 categories; verbal response: 5 categories; total score: 11 categories), and capillary refill rate is encoded as a binary variable. The in-hospital mortality (IHM) prediction and phenotyping tasks follow the standard benchmark settings, with data characteristics summarised in 5. For completeness, we list below the input variables used in MIMIC-III as well as the 25 phenotyping classes. Table 5: Dataset characteristics for MIMIC-III.

# Examples # Time-steps # Features per time-step

MIMIC-III (IHM)

MIMIC-III (Phenotyping)

21,156 48 60

21,728 48 60

29

List of features in MIMIC-III dataset

1. Capillary refill rate-0.0 2. Capillary refill rate-1.0 3. Diastolic blood pressure 4. Fraction inspired oxygen 5. Glascow coma scale eye opening-2 To Pain 6. Glascow coma scale eye opening-3 To speech 7. Glascow coma scale eye opening-1 No Response 8. Glascow coma scale eye opening-4 Spontaneously 9. Glascow coma scale eye opening-0 None 10. Glascow coma scale motor response-1 No Movement 11. Glascow coma scale motor response-3 Abnormal flexion 12. Glascow coma scale motor response-2 Abnormal extension 13. Glascow coma scale motor response-4 Flex-withdraws 14. Glascow coma scale motor response-5 Localizes Pain 15. Glascow coma scale motor response-6 Obeys Commands 16. Glascow coma scale total11

17. Glascow coma scale total10 18. Glascow coma scale total13 19. Glascow coma scale total12 20. Glascow coma scale total15 21. Glascow coma scale total14 22. Glascow coma scale total-3 23. Glascow coma scale total-5 24. Glascow coma scale total-4 25. Glascow coma scale total-7 26. Glascow coma scale total-6 27. Glascow coma scale total-9 28. Glascow coma scale total-8 29. Glascow coma scale verbal response-1 No Response 30. Glascow coma scale verbal response-4 Confused 31. Glascow coma scale verbal response-2 Incomprehensible sounds 32. Glascow coma scale verbal response-3 Inappropriate Words 33. Glascow coma scale verbal response-5 Oriented 34. Glucose 35. Heart Rate

36. Height 37. Mean blood pressure 38. Oxygen saturation 39. Respiratory rate 40. Systolic blood pressure 41. Temperature 42. Weight 43. pH 44. mask-Capillary refill rate 45. mask-Diastolic blood pressure 46. mask-Fraction inspired oxygen 47. mask-Glascow coma scale eye opening 48. mask-Glascow coma scale motor response 49. mask-Glascow coma scale total 50. mask-Glascow coma scale verbal response 51. mask-Glucose 52. mask-Heart Rate 53. mask-Height 54. mask-Mean blood pressure 55. mask-Oxygen saturation 56. mask-Respiratory rate 57. mask-Systolic blood pressure 58. mask-Temperature 59. mask-Weight 60. mask-pH

List of 25 patient disorders involved in Phenotyping in MIMIC-III dataset

1. Acute and unspecified renal failure 2. Acute cerebrovascular disease 3. Acute myocardial infarction 4. Cardiac dysrhythmias 5. Chronic kidney disease 6. Chronic obstructive pulmonary disease 7. Complications of surgical/medical care 8. Conduction disorders 9. Congestive heart failure; non hypertensive

10. Coronary atherosclerosis and related 11. Diabetes mellitus with complications 12. Diabetes mellitus without complication 13. Disorders of lipid metabolism 14. Essential hypertension 15. Fluid and electrolyte disorders 16. Gastrointestinal haemorrhage 17. Hypertension with complications 30

18. Other liver diseases 19. Other lower respiratory disease 20. Other upper respiratory disease 21. Pleurisy; pneumothorax; pulmonary collapse 22. Pneumonia 23. Respiratory failure; insufficiency; arrest 24. Septicemia labour) 25. Shock

(except

in

D

IMPLEMENTATION DETAILS

D.1

Model Architectures

NHS Cohorts and eICU datasets. For tabular datasets, we use a multi-layer perceptron (MLP) (Rumelhart et al., 1986) with a single hidden layer of h units, ReLU activation, and a sigmoid output layer. Dropout (0.25) is applied after the hidden layer. We set h = 256 for eICU and h = 64 for the NHS datasets. To assess cross-architecture generalisation on eICU, we consider additional MLP variants summarised in Table 6, which vary in width and depth while keeping activation functions and regularisation fixed. Table 6: MLP architectures for NHS and eICU datasets. Model

Architecture

Key Details

MLP-1 MLP-2 MLP-3 MLP-4

Wider MLP Wider MLP Deeper MLP Deeper MLP

2h units, ReLU, dropout 0.25 4h units, ReLU, dropout 0.25 2 layers (h, 2h), ReLU, dropout 0.25 3 layers (h, 2h, 4h), ReLU, dropout 0.25

MIMIC-III datasets. For time-series data, we use a temporal convolutional network (TCN) (Bai et al., 2018) as the backbone for dataset condensation. The model consists of a single residual temporal block with 64 channels, kernel size 9, dilation 1, BatchNorm, PReLU activations, and dropout (0.75). The network processes a 48 × 60 multivariate time series, with temporal features mean-pooled and passed through a linear output layer. For in-hospital mortality prediction, the output dimension is 1; for phenotyping, the architecture is unchanged except that the output dimension is 25. For cross-architecture evaluation on the IHM task, we consider additional TCN variants and LSTM baselines summarised in Table 7. The TCN variants extend the base model through multi-scale convolutions and increased depth, while the LSTM models provide a complementary recurrent architecture for comparison. Table 7: Cross-architecture evaluation models on MIMIC-III.

D.2

Model

Architecture

Key Details

TCN-1 TCN-2 LSTM-1 LSTM-2

Multi-scale TCN Multi-scale TCN LSTM LSTM

1 block, kernels [3,5,7], 192 channels, dropout 0.5 2 blocks, kernels [3,5], 256 channels/layer, dropout 0.5, exp. dilation 1 layer, hidden dim 128, unidirectional, dropout 0.25 1 layer, hidden dim 256, unidirectional, dropout 0.25

Teacher Trajectories

For each dataset, we generate 50 teacher trajectories by training the backbone networks from independent random initialisations. MTT trajectories. For MTT, trajectories are generated using the SGD optimiser, with dataset-specific optimisation settings summarised in Table 8. FTD trajectories. For FTD, trajectories are generated using Generalised Sharpness-Aware Minimisation (GSAM). A linear learning rate schedule is used, decaying to zero over tmax optimisation steps. The perturbation radius ρ is coupled to the learning rate and decreases linearly from 1 to 0 over training. Unless otherwise specified, GSAM uses the same base hyperparameters as SGD. DATM trajectories. For DATM, GSAM-based trajectories are used for the tabular datasets (eICU, Oxford, Portsmouth, Birmingham), following the same configuration as FTD. For MIMIC-III (IHM and Phenotyping), we instead use standard SGD trajectories, as GSAM-based trajectories consistently underperform MTT in this structured time-series setting. 31

Table 8: SGD (MTT) teacher trajectory optimisation settings. Dataset

LR

Momentum

Epochs

MIMIC-III (IHM) MIMIC-III (Phenotyping) eICU Oxford Portsmouth Birmingham

0.02 0.05 0.02 0.02 0.01 0.02

0.0 0.9 0.9 0.9 0.9 0.9

60 60 100 100 100 100

Linear and convexified trajectories. For MCT, we learn convexified trajectories from the MTT teacher trajectories using two learned anchor points per trajectory, following the original method. For the convexified linear variant, we retain the same β-projection scheme but restrict the trajectory to a single segment between the initial and final parameters of the teacher trajectory. For the linear trajectory ablation, we use the straight line connecting the initial and final parameters of the corresponding SGD trajectory, without anchor points or β-projection. Bézier trajectories. For BTM, each full teacher trajectory is replaced by a single quadratic Bézier surrogate connecting its initialisation θ0 and final state θT , together with a learnable control point ϕ. The continuous segments used during condensation are sampled from this surrogate, as described in Section 5. The optimisation of ϕ is described below.

D.3

Control Point Optimisation

The control point ϕ is initialised at the midpoint of the trajectory endpoints and optimised to minimise the average training loss along the Bézier curve (Algorithm 2). Since one endpoint corresponds to a random initialisation, this objective does not enforce uniformly low loss across the entire path; instead, it discourages unnecessary excursions and promotes a smoother trajectory with lower average loss between endpoints. Algorithm 2 Control Point Optimisation Require: SGD trajectory endpoints θ0 , θT , dataset D, learning rate ηϕ , convergence tolerance ϵ, maximum iterations Tmax , Monte Carlo samples NM C Ensure: Optimised control point ϕ∗ θ +θ 1: Initialise ϕ ← 0 2 T 2: t ← 0 3: while t < Tmax do MC 4: Sample {ti }N i=1 ∼ U(0, 1) 5: Lavg ← 0 6: for i = 1 to NM C do 7: θti ← (1 − ti )2 θ0 + 2ti (1 − ti )ϕ + t2i θT 8: Sample mini-batch B ⊂ D 1 LCE (fθti , B) 9: Lavg ← Lavg + NM C 10: end for 11: g ← ∇ϕ Lavg 12: if ∥g∥2 < ϵ then 13: break 14: end if 15: ϕ ← ϕ − ηϕ g 16: t←t+1 17: end while 18: return ϕ∗ = ϕ

32

Gradients are computed via automatic differentiation: ∇ϕ Lavg =

where

∂θti ∂ϕ

1

N MC X

NM C i=1

∇θ LCE (fθti , B) ·

∂θti , ∂ϕ

(73)

= 2ti (1 − ti ).

Hyperparameters. We set ηϕ = 10−2 , ϵ = 10−5 , Tmax = 300, and NM C = 5. Computational cost. Control point optimisation requires approximately 1–10 equivalent training epochs per trajectory, depending on dataset size. Specifically, we use 2 epochs for Portsmouth and eICU, 5 for Oxford and Birmingham, and 8 for the MIMIC-III datasets. This cost is incurred once during trajectory construction and amortised over all subsequent condensation iterations. D.4

Dataset Condensation

BTM hyperparameters. Trajectory parameters are selected via validation. We use continuous segments of length ∆t = 0.2, with ts ∼ U(0, 0.8) and te = ts + 0.2. The student performs N = 30 inner-loop updates with learning rate ηs = 0.01. Synthetic data is optimised using SGD with meta learning rate ηx = 100 and momentum 0.9, while ηs is jointly meta-optimised (learning rate 10−4 , momentum 0.5). The batch size is set to b = max(2 × ipc, 256), and condensation runs for Tmax = 20,000 iterations. Baseline configurations. MTT and FTD use discrete trajectory segments of length M = 5 epochs with N = 60 student updates, and share the same synthetic data optimisation settings as BTM. DATM also uses M = 5 and N = 60, but samples segments from an expanding trajectory interval. Let (T − , T, T + ) denote the lower bound, current upper bound, and final upper bound of the sampling range. Segments are initially drawn from [T − , T ] and progressively expanded to [T − , T + ], aligning trajectory difficulty with the synthetic budget. For eICU and NHS datasets:   (0, 20, 40), − + (T , T, T ) = (10, 20, 60),   (20, 40, 100),

50 ipc, 100 ipc, 200, 500 ipc.

For MIMIC-III (IHM and Phenotyping):  (0, 10, 25),    (5, 15, 40), (T − , T, T + ) =  (20, 40, 50),    (30, 40, 60),

50 ipc, 100 ipc, 200 ipc, 500 ipc.

For path complexity ablations (MCT and linear variants), all methods use the same segment length ∆t and number of inner-loop steps N as BTM. D.5

Evaluation Protocol

Model selection during condensation. Synthetic datasets are evaluated every 10 outer iterations by training randomly initialised models from scratch, and selected based on validation AUPRC. We found AUPRC to be a more reliable selection metric than AUROC, as high AUPRC consistently corresponded to strong AUROC, but not vice versa. Evaluation settings during condensation are summarised in Table 9. Final evaluation. Final results are obtained by retraining models from scratch on the selected synthetic datasets using longer training schedules (Table 10). 33

Table 9: Evaluation settings during condensation. Dataset

LR

Momentum

Epochs

MIMIC-III (IHM) MIMIC-III (Phenotyping) eICU, Oxford, Portsmouth, Birmingham

0.02 0.05 0.05

0.9 0.9 0.9

60 60 50

Table 10: Final evaluation settings. Dataset

LR

Momentum

Epochs

MIMIC-III (IHM) MIMIC-III (Phenotyping) eICU, Oxford, Portsmouth, Birmingham

0.02 0.05 0.05

0.9 0.9 0.9

80 80 100

Cross-architecture evaluation. We evaluate generalisation using alternative architectures. For eICU, we use the MLP variants described in Section D.1. For MIMIC-III, we evaluate TCN and LSTM variants. Within each model family, we use the same optimisation settings as the corresponding base architecture. For LSTM models on MIMIC-III, we instead use a learning rate of 0.01, momentum 0.9, and 30 epochs.

E

Additional Experiments

E.1

Inner Loop Steps

The inner loop controls the number of gradient updates the student model takes when trained on the synthetic data before matching against the expert trajectory. In standard TM, this is tied to the expert optimisation epochs M, but with Bézier surrogates the continuous parameterisation t ∈ [0, 1] decouples segment length (tstart , tend ) from discrete optimisation steps. As a result, the optimal number of steps must be determined empirically. Figure 7 ablates N at 200 ipc. While prior methods typically require N ≥ 40 (Guo et al., 2024), BTM sustains strong AUROC and AUPRC even at N = 30, demonstrating that Bézier surrogates provide stable supervision, enabling the student to train effectively with few gradient steps. (a) PUH

0.904

(b) eICU

0.902 0.901

0.838

0.859

AUROC

AUROC

AUROC

0.903

0.858

30

40

50

60

70

# Inner Loop Steps

80

0.857

0.835 0.833

0.858

0.900

(c) MIMIC-III

0.840

0.860

0.830 30

40

50

60

70

# Inner Loop Steps

80

30

40

50

60

70

# Inner Loop Steps

80

Figure 7: Impact of inner-loop steps N on AUROC performance at 200 ipc. BTM achieves strong performance with only 30 steps, reducing computational overhead. Similar trends observed for AUPRC.

E.2

Initialisation Strategy for the Synthetic Dataset

Table 11 examines the impact of synthetic-data initialisation on condensation performance in eICU. We compare real initialisation, which seeds synthetic inputs from real training samples, with random initialisation, which samples inputs from class-conditional Gaussian distributions. The two strategies yield broadly 34

Table 11: Initialisation strategy comparison on eICU. Random initialisation remains competitive while providing stronger privacy guarantees. IPC

Init. 50

100

200

500

AUROC

Real Random

0.854±0.002 0.852±0.009

0.859±0.001 0.858±0.004

0.861±0.001 0.853±0.004

0.874±0.002 0.862±0.002

AUPRC

Real Random

0.479±0.002 0.463±0.015

0.486±0.001 0.474±0.007

0.476±0.001 0.475±0.008

0.506±0.001 0.499±0.004

comparable performance, although real initialisation provides a modest advantage at 500 ipc. These results suggest that BTM is not strongly dependent on direct seeding from real examples, and that competitive condensation can still be achieved from a purely synthetic starting point. This comparison is important because the initialisation scheme affects how directly the optimisation begins from the observed data distribution. Real initialisation can place synthetic inputs closer to the data manifold at the outset and may therefore improve optimisation, whereas random initialisation provides a weaker direct dependence on individual training examples at initialisation. However, formal privacy guarantees do not follow from the initialisation strategy alone. Rather, they arise from applying differential privacy to the condensation pipeline itself. Under such a mechanism, both real and random initialisation can be used within a privacy-preserving framework, while random initialisation may still offer a conceptually cleaner starting point with reduced direct dependence on observed samples. E.3

Teacher Segment Length in BTM

Table 12 summarises the effect of teacher segment length, ∆t = te − ts , on the Portsmouth dataset. When ∆t is too small, the induced target displacement is overly restrictive; when ∆t is too large, the corresponding displacement exceeds what the student can reliably realise within its optimisation budget. We find that ∆t = 0.2 offers the best balance between these two effects, and accordingly use this value in all experiments, as discussed earlier. Table 12: Impact of continuous segment length (∆t) on AUPRC for the Portsmouth dataset. A segment length of 0.2 provides the optimal look-ahead window for trajectory alignment.

IPC

Segment Length (∆t) 0.1 0.2 0.3 0.4

100

200

500

0.558±0.012 0.572±0.001 0.564±0.010 0.556±0.011

0.582±0.007 0.596±0.001 0.585±0.008 0.579±0.009

0.589±0.007 0.609±0.001 0.593±0.005 0.581±0.004

35

Record · ID 126528 · SHA-256 a7004b7551fd07d0
Retrieved via Conceptio — every document is proof-bundled with source, license, and retrieval metadata.