Changing Model Behavior at Test-Time Using Reinforcement Learning
Not on the map. Why?
Abstract (arXiv)
Machine learning models are often used at test-time subject to constraints and trade-offs not present at training-time. For example, a computer vision model operating on an embedded device may need to perform real-time inference, or a translation model operating on a cell phone may wish to bound its average compute time in order to be power-efficient. In this work we describe a mixture-of-experts model and show how to change its test-time resource-usage on a per-input basis using reinforcement learning. We test our method on a small MNIST-based example.
Key facts. Cited by 15 papers in the map (data through 7 Oct 2026). First posted on arXiv 24 Feb 2017.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 3; 2022: 2; 2023: 3; 2024: 3; 2025: 1; 2026: 3.
15 papers.
Sort:
By area:
- 2021Dynamic Neural Networks: A Surveycited by 405
- 2021AdaViT: Adaptive Tokens for Efficient Vision Transformercited by 360
- 2021Pruning and Quantization for Deep Neural Network Acceleration: A Surveycited by 259
- 2022Controllable Dynamic Multi-Task Architecturescited by 21
- 2022Task-Customized Self-Supervised Pre-training with Scalable Dynamic Routingcited by 18
- 2023ScrollNet: Dynamic Weight Importance for Continual Learningcited by 4
- 2023FIANCEE: Faster Inference of Adversarial Networks via Conditional Early Exitscited by 3
- 2025Energy-Aware Deep Learning on Resource-Constrained Hardwarecited by 3
- 2023Temporal Patience: Efficient Adaptive Deep Learning for Embedded Radar Data Processingcited by 2
- 2024Temporal Decisions: Leveraging Temporal Correlation for Efficient Decisions in Early Exit Neural Networkscited by 1
- 2024Predicting Probabilities of Error to Combine Quantization and Early Exiting: QuEEcited by 1
- 2026TTQ: Activation-Aware Test-Time Quantization to Accelerate LLM Inference On The Flycited by 1
3 more
- 2024Efficient Post-Training Augmentation for Adaptive Inference in Heterogeneous and Distributed IoT Environmentscited by 0
- 2026HERCULES: Hardware-Efficient, Robust, Continual Learning Neural Architecture Searchcited by 0
- 2026Exploring Budgeted Image Classification with Content-Sensitive Resource Allocationcited by 0
- 2026Exploring Budgeted Image Classification with Content-Sensitive Resource Allocation27 Jul 2026
- 2026HERCULES: Hardware-Efficient, Robust, Continual Learning Neural Architecture Search3 May 2026
- 2026TTQ: Activation-Aware Test-Time Quantization to Accelerate LLM Inference On The Fly11 Mar 2026
- 2025Energy-Aware Deep Learning on Resource-Constrained Hardware18 May 2025
- 2024Predicting Probabilities of Error to Combine Quantization and Early Exiting: QuEE20 Jun 2024
- 2024Temporal Decisions: Leveraging Temporal Correlation for Efficient Decisions in Early Exit Neural Networks12 Mar 2024
- 2024Efficient Post-Training Augmentation for Adaptive Inference in Heterogeneous and Distributed IoT Environments12 Mar 2024
- 2023Temporal Patience: Efficient Adaptive Deep Learning for Embedded Radar Data Processing11 Sep 2023
- 2023ScrollNet: Dynamic Weight Importance for Continual Learning31 Aug 2023
- 2023FIANCEE: Faster Inference of Adversarial Networks via Conditional Early Exits20 Apr 2023
- 2022Task-Customized Self-Supervised Pre-training with Scalable Dynamic Routing26 May 2022
- 2022Controllable Dynamic Multi-Task Architectures28 Mar 2022
3 more
- 2021AdaViT: Adaptive Tokens for Efficient Vision Transformer14 Dec 2021
- 2021Dynamic Neural Networks: A Survey9 Feb 2021
- 2021Pruning and Quantization for Deep Neural Network Acceleration: A Survey24 Jan 2021
- 2021Dynamic Neural Networks: A Surveycited by 405
- 2021AdaViT: Adaptive Tokens for Efficient Vision Transformercited by 360
- 2021Pruning and Quantization for Deep Neural Network Acceleration: A Surveycited by 259
- 2022Controllable Dynamic Multi-Task Architecturescited by 21
- 2022Task-Customized Self-Supervised Pre-training with Scalable Dynamic Routingcited by 18
- 2023ScrollNet: Dynamic Weight Importance for Continual Learningcited by 4
- 2025Energy-Aware Deep Learning on Resource-Constrained Hardwarecited by 3
- 2023Temporal Patience: Efficient Adaptive Deep Learning for Embedded Radar Data Processingcited by 2
- 2024Temporal Decisions: Leveraging Temporal Correlation for Efficient Decisions in Early Exit Neural Networkscited by 1
- 2026TTQ: Activation-Aware Test-Time Quantization to Accelerate LLM Inference On The Flycited by 1
- 2024Efficient Post-Training Augmentation for Adaptive Inference in Heterogeneous and Distributed IoT Environmentscited by 0
- 2026HERCULES: Hardware-Efficient, Robust, Continual Learning Neural Architecture Searchcited by 0
12 most cited of 13 All 13 (JSON)
- 2026Exploring Budgeted Image Classification with Content-Sensitive Resource Allocation27 Jul 2026
- 2026HERCULES: Hardware-Efficient, Robust, Continual Learning Neural Architecture Search3 May 2026
- 2026TTQ: Activation-Aware Test-Time Quantization to Accelerate LLM Inference On The Fly11 Mar 2026
- 2025Energy-Aware Deep Learning on Resource-Constrained Hardware18 May 2025
- 2024Temporal Decisions: Leveraging Temporal Correlation for Efficient Decisions in Early Exit Neural Networks12 Mar 2024
- 2024Efficient Post-Training Augmentation for Adaptive Inference in Heterogeneous and Distributed IoT Environments12 Mar 2024
- 2023Temporal Patience: Efficient Adaptive Deep Learning for Embedded Radar Data Processing11 Sep 2023
- 2023ScrollNet: Dynamic Weight Importance for Continual Learning31 Aug 2023
- 2022Task-Customized Self-Supervised Pre-training with Scalable Dynamic Routing26 May 2022
- 2022Controllable Dynamic Multi-Task Architectures28 Mar 2022
- 2021AdaViT: Adaptive Tokens for Efficient Vision Transformer14 Dec 2021
- 2021Dynamic Neural Networks: A Survey9 Feb 2021
12 newest of 13 All 13 (JSON)
Similar, not cited
Close in wording, no citation either way
- 2026A Survey on Self-Improving Test-Time Intelligence: Feedback-Driven Adapting, Learning, and Scaling at Inference
- 2026Temporally Extended Mixture-of-Experts Models
- 2020Contextual Policy Transfer in Reinforcement Learning Domains via Deep Mixtures-of-Experts
- 2026On-the-Fly VLA Adaptation via Test-Time Reinforcement Learning
- 2025Scaling Test-Time Compute Without Verification or RL is Suboptimal
- 2026TTA-Vid: Generalized Test-Time Adaptation for Video Reasoning
- 2026Harnessing Image Question Dependence for Better VLM Test-time Reinforcement Learning
- 2025Local Mixtures of Experts: Essentially Free Test-Time Training via Model Merging