CASTER Enables Gradient-Free Test-Time Adaptation for Frozen Models

Salim Khazem, Ibrahim Mohamed Serouis· September 2, 2026 View original

Key takeaways

  • CASTER enables gradient-free test-time adaptation for frozen AI models.
  • It is ideal for resource-constrained environments and inference-only accelerators.
  • The method analytically transports source class distributions using affine transformations.
  • An accompanying certificate helps assess and ensure the reliability of the adaptation process.

Who benefits

Edge AIAutomotiveIoTManufacturingHealthcare

Summary

Researchers introduce CASTER, a gradient-free method for test-time adaptation (TTA) that works with frozen models, inference-only accelerators, and memory-constrained deployments. CASTER analytically transports source class distributions using affine transformations estimated from target-batch moments, outperforming k-NN on frozen features with significantly less state and offering a transportability certificate to gauge reliability.

A new research paper presents CASTER, a novel gradient-free approach to Test-Time Adaptation (TTA) designed for scenarios where model parameters cannot be updated during inference. This method is particularly beneficial for deployments on inference-only accelerators, with frozen or third-party models, or in memory-constrained environments, overcoming limitations of traditional gradient-based TTA. CASTER operates by storing source class statistics within a discriminative subspace and then estimating a class-shared affine transformation from the moments of the target batch. It analytically transports these source class distributions before classification, requiring no backward pass, optimizer state, or stored source feature bank. The method consistently outperforms k-NN on identical frozen features across various datasets and backbones, while maintaining significantly less state. The paper also introduces an empirical residual-to-margin transportability certificate to assess the reliability of the affine transport, especially in challenging scenarios like ImageNet-C where unconditional transport can lead to significant performance drops. This certificate helps gate the adaptation process, converting potential performance losses into gains and providing a safety signal for deployment.

Why it matters

For professionals deploying AI models in resource-constrained or sensitive environments, CASTER offers a practical solution for adapting models to new data distributions without retraining or modifying core parameters. This improves model robustness and extends the lifespan of deployed models.

How to implement this in your domain

  1. 1Evaluate CASTER for existing frozen or third-party AI models deployed in production to improve their performance on new data distributions.
  2. 2Integrate the empirical residual-to-margin transportability certificate to monitor and gate adaptation, ensuring reliable performance.
  3. 3Consider CASTER for edge AI deployments or applications on specialized inference hardware where gradient-based adaptation is not feasible.
  4. 4Benchmark CASTER against other lightweight adaptation techniques to determine its suitability for specific use cases.

Original post by Salim Khazem, Ibrahim Mohamed Serouis

"arXiv:2609.00374v1 Announce Type: new Abstract: Test-time adaptation (TTA) typically assumes that model parameters can be updated at inference time. This assumption is restrictive for inference-only accelerators, frozen or third-party models, and memory-constrained deployments, a…"

View on X

Originally posted by Salim Khazem, Ibrahim Mohamed Serouis on X · view source

Want to go deeper?

Turn these trends into skills with Learnijoy's hands-on AI & tech courses.

Explore courses