Papers
Topics
Authors
Recent
Search
2000 character limit reached

Score-Matched Actor-Critic (SMAC)

Updated 5 July 2026
  • Score-Matched Actor-Critic (SMAC) is a reinforcement learning method that aligns the policy's score with the critic’s action-gradient for efficient, optimal updates.
  • It employs a diffusion-based actor in continuous control and regularizes the critic to ensure robust offline-to-online transfer without initial performance drops.
  • Empirical results demonstrate that SMAC improves sample efficiency and exploration while smoothing the transition from offline training to effective online fine-tuning.

Score-Matched Actor-Critic (SMAC) denotes actor-critic methods built around a first-order correspondence between a policy score and the action-gradient of a critic. In the diffusion-policy formulation of "Learning a Diffusion Model Policy from Rewards via Q-Score Matching" (Psenka et al., 2023), the method is also called Q-Score Matching (QSM) and is developed for off-policy reinforcement learning in continuous action spaces. In "SMAC: Score-Matched Actor-Critics for Robust Offline-to-Online Transfer" (Lara et al., 19 Feb 2026), the same acronym denotes an offline reinforcement learning method that regularizes the critic so that offline optima transfer to online value-based fine-tuning without an initial drop in performance. Across both uses, the core premise is that actor-critic compatibility can be expressed through score-gradient alignment rather than through a simple behavior cloning term or unrestricted gradient-based fine-tuning.

1. Naming, scope, and problem setting

The term SMAC is used for two related constructions. The 2023 formulation addresses off-policy reinforcement learning with a diffusion-model actor in continuous domains, and the 2026 formulation addresses robust offline-to-online transfer in offline reinforcement learning. Both works center the relation between a policy score and aQ\nabla_a Q, but they instantiate that relation differently (Psenka et al., 2023, Lara et al., 19 Feb 2026).

Work Setting Core mechanism
(Psenka et al., 2023) Off-policy RL with a diffusion-model policy Match the denoising score network to aQϕ(s,a)\nabla_a Q_\phi(s,a)
(Lara et al., 19 Feb 2026) Offline RL for offline-to-online transfer Regularize aQϕ(s,a)\nabla_a Q_\phi(s,a) to match a scaled dataset score estimate

In the diffusion-policy setting, the motivation is that previous works with diffusion actors fail to exploit the score-based structure of diffusion models and instead utilize a simple behavior cloning term to train the actor, limiting their ability in the actor-critic setting. In the offline-to-online transfer setting, the motivation is different: modern offline RL methods can learn performant actor-critics, but fine-tuning them online with value-based RL algorithms typically causes immediate drops in performance. The 2026 work provides evidence consistent with the hypothesis that offline maxima for prior algorithms and online maxima are separated by low-performance valleys that gradient-based fine-tuning traverses (Lara et al., 19 Feb 2026).

2. Diffusion-policy formulation and denoising score networks

In the Q-Score Matching formulation, the stochastic policy πθ(as)\pi_\theta(a|s) is represented in continuous action space A=Rd\mathcal A=\mathbb R^d as a KK-step variance-preserving diffusion model conditioned on state ss (Psenka et al., 2023). The forward noising schedule is fixed as β1,,βK(0,1)\beta_1,\ldots,\beta_K\in(0,1), with transitions

q(a1s)=N(a1;0,I),q(a^1|s)=\mathcal N(a^1;0,I),

and

q(atat1,s)=N ⁣(at;1βtat1,βtI),t=2,,K.q(a^t|a^{t-1},s)=\mathcal N\!\bigl(a^t;\sqrt{1-\beta_t}\,a^{t-1},\beta_t I\bigr),\qquad t=2,\ldots,K.

A denoising score network is then learned as

aQϕ(s,a)\nabla_a Q_\phi(s,a)0

The final action aQϕ(s,a)\nabla_a Q_\phi(s,a)1 is sampled by running the standard reverse SDE or its discretization, for example Langevin dynamics or ancestral sampling. In this construction, the score of the marginal policy with respect to aQϕ(s,a)\nabla_a Q_\phi(s,a)2 is exactly

aQϕ(s,a)\nabla_a Q_\phi(s,a)3

This identity is the technical basis for replacing direct differentiation through the entire diffusion-sampling process with score matching against the critic. The paper emphasizes that the resulting converged policies are implicitly multi-modal and explorative in continuous domains, which is a direct consequence of using a diffusion actor rather than a unimodal Gaussian parameterization (Psenka et al., 2023).

3. First-order optimality and score-gradient alignment

The theoretical core of the 2023 formulation is a link between the structure of the policy score and the action-gradient of the Q-function. At the global optimum of the expected discounted return

aQϕ(s,a)\nabla_a Q_\phi(s,a)4

any maximizing policy aQϕ(s,a)\nabla_a Q_\phi(s,a)5 must satisfy, for all aQϕ(s,a)\nabla_a Q_\phi(s,a)6 in support where aQϕ(s,a)\nabla_a Q_\phi(s,a)7,

aQϕ(s,a)\nabla_a Q_\phi(s,a)8

The paper outlines the key derivation in the deterministic setting through the path-value

aQϕ(s,a)\nabla_a Q_\phi(s,a)9

and a perturbation argument on the actor’s vector field aQϕ(s,a)\nabla_a Q_\phi(s,a)0. A Taylor expansion around aQϕ(s,a)\nabla_a Q_\phi(s,a)1 shows that if the perturbed drift acquires a component along aQϕ(s,a)\nabla_a Q_\phi(s,a)2 that the original drift lacked, then the path value increases. Repeating this argument throughout state-action space yields the stationary condition that the actor score must be parallel to the action-gradient of the critic (Psenka et al., 2023).

The 2026 formulation states the corresponding identity in maximum-entropy reinforcement learning as

aQϕ(s,a)\nabla_a Q_\phi(s,a)3

and approximately enforces during training

aQϕ(s,a)\nabla_a Q_\phi(s,a)4

Taken together, these two formulations distinguish between a support-wise collinearity condition and an exact scaled equality. The 2023 paper further states that any local improvement step that better aligns aQϕ(s,a)\nabla_a Q_\phi(s,a)5 with aQϕ(s,a)\nabla_a Q_\phi(s,a)6 strictly increases the expected return aQϕ(s,a)\nabla_a Q_\phi(s,a)7, and that the only fixed points satisfy

aQϕ(s,a)\nabla_a Q_\phi(s,a)8

on the support of the policy. Under mild regularity, the resulting stationary policy coincides with the Boltzmann policy aQϕ(s,a)\nabla_a Q_\phi(s,a)9 in the continuous-time limit (Psenka et al., 2023).

4. Losses, updates, and implementation

In Q-Score Matching, the policy update is defined directly in terms of score matching against the critic. At each diffusion time-step πθ(as)\pi_\theta(a|s)0, the noised action is

πθ(as)\pi_\theta(a|s)1

and the actor objective is

πθ(as)\pi_\theta(a|s)2

with weighting factors πθ(as)\pi_\theta(a|s)3 that are often set to πθ(as)\pi_\theta(a|s)4. In practice, the method maintains a replay buffer πθ(as)\pi_\theta(a|s)5 of transitions πθ(as)\pi_\theta(a|s)6. The critic is updated by TD learning, for example in TD3 or SAC style, using target

πθ(as)\pi_\theta(a|s)7

where πθ(as)\pi_\theta(a|s)8 is sampled from the current diffusion actor, and minimizing

πθ(as)\pi_\theta(a|s)9

The actor update samples a minibatch A=Rd\mathcal A=\mathbb R^d0, generates noised versions A=Rd\mathcal A=\mathbb R^d1, computes A=Rd\mathcal A=\mathbb R^d2 by automatic differentiation of the frozen critic, and minimizes

A=Rd\mathcal A=\mathbb R^d3

All gradients in A=Rd\mathcal A=\mathbb R^d4 flow only through the score network A=Rd\mathcal A=\mathbb R^d5; the Q-gradient is treated as a fixed target. The paper’s pseudocode initializes critics A=Rd\mathcal A=\mathbb R^d6, target critics A=Rd\mathcal A=\mathbb R^d7, a score network A=Rd\mathcal A=\mathbb R^d8, and a replay buffer, then alternates data collection, critic updates, actor score updates, and Polyak target updates (Psenka et al., 2023).

The 2026 SMAC formulation modifies only the offline phase of an otherwise standard actor-critic pipeline. It estimates the dataset score A=Rd\mathcal A=\mathbb R^d9 via a return-conditioned diffusion model KK0, introduces a small network KK1 to scale that estimate, and defines the score-matching regularizer

KK2

where KK3 is a 50/50 mix of actions from KK4 and uniform random actions. The full critic loss is

KK5

while the actor remains unchanged from SAC with loss KK6. In practice the method also uses target networks KK7 and ensemble critics as in SAC. The reported offline pseudocode updates KK8, KK9, and ss0 with the Muon optimizer, described as a spectral-norm-aware optimizer, and reports typical hyperparameters ss1, ss2, ss3, ss4, batch size ss5, ensembles of ss6 critics, and target ss7 (Lara et al., 19 Feb 2026).

5. Empirical findings

The diffusion-policy Q-Score Matching paper benchmarks SMAC on six DeepMind Control Suite tasks, including Walker Walk and Cheetah Run, and compares it to TD3 and SAC. The reported findings are threefold: sample efficiency matches or outperforms SAC and TD3, especially in the low-data regime; the learned diffusion actor naturally represents multi-modal action distributions, with the paper citing two swing-up modes in cartpole as an example; and exploration improves because sampling from the full diffusion chain, with optional Gaussian noise, visits diverse actions that Gaussian actors cannot. Figure 1 in the paper shows cumulative returns versus environment steps, and Figure 2 illustrates policy samples for a toy swing-up (Psenka et al., 2023).

The offline-to-online transfer paper evaluates SMAC on six D4RL tasks and reports smooth transfer to Soft Actor-Critic and TD3 in 6/6 tasks. In 4/6 environments, it reduces regret by 34–58% over the best baseline. It also reports that all six SMAC pre-trained agents exhibit no initial drop when fine-tuned with SAC or TD3, whereas CalQL, IQL, and TD3+BC often suffer a 5–20% collapse before recovering (Lara et al., 19 Feb 2026).

Environment Best baseline ss8 SMAC ss9
door 120.0 β1,,βK(0,1)\beta_1,\ldots,\beta_K\in(0,1)0 50.3 β1,,βK(0,1)\beta_1,\ldots,\beta_K\in(0,1)1
hopper 293.7 β1,,βK(0,1)\beta_1,\ldots,\beta_K\in(0,1)2 386.3 β1,,βK(0,1)\beta_1,\ldots,\beta_K\in(0,1)3 (SMAC wins final)
kitchen 467.1 β1,,βK(0,1)\beta_1,\ldots,\beta_K\in(0,1)4 131.4 β1,,βK(0,1)\beta_1,\ldots,\beta_K\in(0,1)5
pen 8.0 β1,,βK(0,1)\beta_1,\ldots,\beta_K\in(0,1)6 5.3 β1,,βK(0,1)\beta_1,\ldots,\beta_K\in(0,1)7
relocate 98.1 β1,,βK(0,1)\beta_1,\ldots,\beta_K\in(0,1)8 62.8 β1,,βK(0,1)\beta_1,\ldots,\beta_K\in(0,1)9
walker2d 650.5 q(a1s)=N(a1;0,I),q(a^1|s)=\mathcal N(a^1;0,I),0 650.5 q(a1s)=N(a1;0,I),q(a^1|s)=\mathcal N(a^1;0,I),1 (tied)

These empirical results support two distinct claims. In the 2023 formulation, score matching enables expressive and exploratory continuous-control actors without differentiating through the entire diffusion sampler. In the 2026 formulation, score matching is used to reshape the offline objective so that first-order optimization can move from offline optima toward better online optima without traversing an initial low-reward region (Psenka et al., 2023, Lara et al., 19 Feb 2026).

6. Interpretation, misconceptions, and extensions

A central distinction between the two SMAC formulations is where score matching enters the actor-critic loop. In the diffusion-policy formulation, score matching is the actor update itself: the denoising score network is directly matched to the critic’s action-gradient. In the offline-to-online transfer formulation, the actor remains a Gaussian policy with the standard SAC actor loss, and score matching appears as a critic regularizer during offline pre-training (Psenka et al., 2023, Lara et al., 19 Feb 2026).

This distinction resolves a common naming ambiguity. SMAC is not a single fixed algorithmic template; it is a label applied to two methods that share a derivative-matching principle. It is also not equivalent to adding a behavior cloning term to actor-critic learning. The 2023 paper explicitly contrasts its method with prior diffusion-policy approaches that utilize a simple behavior cloning term to train the actor, and argues that this limits their ability in the actor-critic setting. Likewise, the diffusion-policy SMAC update does not require backpropagating through the entire diffusion-sampling process, because it only needs to differentiate through the denoising model rather than the entire diffusion model evaluation (Psenka et al., 2023).

The offline-to-online transfer work interprets performance valleys as a loss-landscape phenomenon: offline maxima for prior algorithms and online maxima lie in different basins, and straight-gradient fine-tuning paths cross low-reward regions. Its informal proposition is that, under SMAC regularization, there exists a path q(a1s)=N(a1;0,I),q(a^1|s)=\mathcal N(a^1;0,I),2 along which q(a1s)=N(a1;0,I),q(a^1|s)=\mathcal N(a^1;0,I),3 is monotonic. This suggests that the regularizer is intended not merely to improve offline return, but to alter the topology of the solution set relevant to subsequent online optimization (Lara et al., 19 Feb 2026).

The extensions explicitly proposed for the 2026 formulation include pairing the offline regularizer with SAC, TD3, TD3+BC, or AWR; applying it in vision-based or language-conditioned reinforcement learning by training a diffusion-based score model on mixed modalities; and combining the SMAC regularizer with alternative pessimism techniques such as CQL. A plausible implication is that score-gradient alignment can be regarded as a broader design principle for actor-critic compatibility across expressive policy classes and across offline, online, and transfer regimes (Lara et al., 19 Feb 2026).

Definition Search Book Streamline Icon: https://streamlinehq.com
References (2)

Topic to Video (Beta)

No one has generated a video about this topic yet.

Whiteboard

No one has generated a whiteboard explanation for this topic yet.

Follow Topic

Get notified by email when new papers are published related to Score Matched Actor-Critic (SMAC).