AI RESEARCH

Precision autotuning for linear solvers via contextual bandit-based RL

arXiv CS.LG

ArXi:2601.00728v4 Announce Type: replace We propose a reinforcement learning (RL) framework for adaptive precision tuning for linear solvers, which can be extended to general algorithms. The framework is formulated as a contextual bandit problem and solved using incremental action-value estimation with a discretized state space to select optimal precision configurations for computational steps, balancing precision and computational efficiency. To verify its effectiveness, we apply the framework to iterative refinement for solving linear systems $Ax = b.