atlas

Research map / Learning theory and algorithms / Combinatorial and search optimization

Learning to Relax: Setting Solver Parameters Across a Sequence of Linear System Instances

ICLR 2024 · Learning theory and algorithms › Combinatorial and search optimization · approach: Constrained neural network optimization · 0 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Shows bandit and contextual bandit online learning can sequentially tune the SOR relaxation parameter (and CG/SSOR variants) across linear systems, with regret guarantees using only iteration counts as feedback.

Problem

Choosing the relaxation parameter omega of SOR (and related solvers like preconditioned CG) is instance-dependent and expensive to optimize, yet many related linear systems are solved in sequence; the goal is end-to-end learning guarantees for tuning these parameters.

Approach

Treat omega selection as a bandit problem with iteration count as feedback; bound the number of iterations by a learnable surrogate U(omega) based on the asymptotic convergence rate, and run Tsallis-INF for a fixed comparator, or discretized Tsallis-INF / a Chebyshev-regression SquareCB for instance-optimal policies in the diagonally shifted setting.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use contextual bandit (ChebCB, SquareCB with Chebyshev regression) instead of discretized Tsallis-INF for diagonal shiftsSSOR parameter selection in diagonally shifted setting · semi-stochastic diagonal shiftsregret w.r.t. instance-optimal policy (theoretical exponent): O(T^{9/11}sqrt n) vs O(T^{3/4}) worse bound, but converges to instance-optimal policy faster empiricallyMixed

Limitations

Experimental setup

Datasets: discrete Laplacian of 100x100 square domain (synthetic) 2D heat equation with time-varying diffusion coefficient (synthetic)

Benchmarks: synthetic diagonally shifted SOR instances 2D heat equation SSOR-preconditioned CG

Metrics: total solver iterations regret

Baselines: best fixed omega instance-optimal policy Tsallis-INF randomized block-Krylov spectral radius approximation Exp3

Builds on: Tsallis-INF SquareCB algorithms with predictions Exp3

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy