Parameter-Efficient Temporal Domain Generalization
Organizations: The University of Tokyo, Tokyo, Japan · LocationMind, Tokyo, Japan · Emory University, Atlanta, GA, USA · Jilin University, Changchun, China
Abstract
Temporal domain generalization (TDG) learns from a sequence of observed domains to predict models for unseen future domains. Existing parameter-dynamics methods become prohibitively expensive at modern backbone scales because their temporal encoders and generators operate on the entire adapted parameter state. Replacing full-model updates with LoRA does not remove this failure mode: the dynamics model must still encode and reconstruct every adapter parameter, making the auxiliary model itself prohibitively large for modern backbones. We propose Low-rank Time-varying Core (LoTiC), a parameter-efficient TDG method that decomposes temporal updates into shared low-rank bases and a compact core. The bases define a stable update subspace, while only the core evolves. LoTiC therefore learns temporal dynamics in an r*r space instead of tracking the full adapter. We justify this factorization through the geometry of parameter increments and support continuous, recurrent, and time-conditioned core models. Experiments span six controlled and real-world benchmarks, three data modalities, classification and generation, and backbones from 66M to 8B parameters. LoTiC ranks first in 14 of 15 settings, including near-perfect future-domain accuracy with an 8B backbone on Rotating 2-Moons. It adds only 0.13--0.51% trainable parameters beyond LoRA and trains 1.8--7.1 faster than parameter-dynamics baselines. These results establish that temporal parameter dynamics can remain both learnable and efficient as pretrained backbones scale.
Figures & tables
| Dataset | Backbone Scale | ERM | Last Domain | IncFT | DRAIN-L | Koodos-L | FreKoo-L | Prompt TDG | TEA | LoTiC |
| Rot-2- Moons | DistilBERT 66M | 18.5 (1.0) | 12.7 (0.9) | 28.2 (6.5) | 21.4 (5.9) | 11.9 (0.5) | 11.8 (1.1) | 25.9 (4.2) | 21.8 (3.5) | 0.6 (0.3) |
| Qwen3 0.6B | 23.5 (4.4) | 13.7 (1.3) | 24.7 (1.0) | OOM | OOM | OOM | 49.8 (15.9) | 25.6 (1.3) | 2.4 (2.5) | |
| LLaMA-3 8B | 21.8 (6.1) | 12.7 (0.6) | 22.1 (7.2) | OOM | OOM | OOM | 23.4 (5.4) | 17.3 (1.2) | 0.9 (0.7) | |
| Rot- MNIST | DINOv2-B 86M | 20.0 (1.2) | 45.9 (2.0) | 44.7 (2.0) | 61.8 (2.5) | 61.3 (5.2) | 28.7 (1.0) | 19.1 (1.4) | 18.7 (0.6) | 8.6 (2.4) |
| DINOv2-L 0.3B | 16.6 (1.9) | 37.5 (3.3) | 37.8 (2.7) | OOM | OOM | OOM | 15.5 (1.1) | 15.6 (0.4) | 7.3 (1.0) |
| Method | Total params | Trainable params | Trainable vs. LoRA | Ckpt. (MB) | Train (s) | Test (s) | Test err. (%) |
| ERM | 66.59M | 225,026 | 1.00 | 0.900 | 10 | 18.5 | |
| DRAIN–LoRA | 182.01M | 115.64M | 513.91 | 462.6 | 165.3 | 21.4 | |
| Koodos–LoRA | 532.0M | 465.65M | 2069.30 | 1862.6 | 580 | 11.9 | |
| FreKoo–LoRA | 763.0M | 696.67M | 3095.95 | 2786.7 | 150 | 11.8 | |
| LoTiC | 66.59M | 226,178 | 1.01 | 0.905 | 82 | 0.6 |
| Backbone | LoRA | LoTiC | Additional |
| DistilBERT (66M) | 225,026 | 226,178 | +1,152 (0.51%) |
| DINOv2-B (86M) | 450,058 | 452,362 | +2,304 (0.51%) |
| DINOv2-L (0.3B) | 1,189,898 | 1,194,506 | +4,608 (0.39%) |
| Qwen3 (0.6B) | 1,610,754 | 1,616,130 | +5,376 (0.33%) |
| LLaMA-3 (8B) | 4,739,074 | 4,745,218 | +6,144 (0.13%) |
Appendix figures & tables2 assets
Supplementary material from the paper’s appendix.
Appendix
| Benchmark | Backbone | Targets | Precision | LR | Epochs | Batch | ||
| Rotating 2-Moons | DistilBERT | 8 | 64 | FP32 | 200 | 200 | ||
| Qwen3–0.6B | 8 | 64 | FP32 | 500 | 200 | |||
| LLaMA-3–8B | 8 | 128 | BF16/FP32 | 500 | 200 | |||
| Rotated MNIST | DINOv2-B/L | 8 | 64 | FP32 | 300 | 200 | ||
| Yearbook | DINOv2-B/L | 8 | 32 | FP32 | 100 | 128 | ||
| HuffPost | DistilBERT | 8 | 64 | FP32 | 10 | 128 |
| Method | Evolved state | Auxiliary parameters | Training trajectory |
| DRAIN–LoRA | |||
| Koodos–LoRA | |||
| FreKoo–LoRA | |||
| LoTiC (Flow) | per matrix |
Domain Generalization
Domain generalization develops models that perform reliably in new environments without training on target-domain data.
See all 375 papers