Papers
Topics
Authors
Recent
Search
2000 character limit reached

FMR: Feedback Manipulation Regularization

Updated 11 July 2026
  • Feedback Manipulation Regularization (FMR) is a framework that integrates evaluative feedback into offline imitation learning to directly reshape policy distributions for human-aligned behavior.
  • It employs per-action temperature adjustments and a reverse-KL regularizer to modulate policy entropy based on feedback polarity, ensuring effective alignment from mixed-quality demonstrations.
  • FMR operates in a single offline training stage without environment interaction, achieving significant misalignment reductions in safety-critical benchmarks while preserving performance.

Feedback Manipulation Regularization (FMR) is a regularization framework for offline imitation learning that uses offline evaluative feedback to directly reshape an imitation-learned policy toward human-aligned behavior in fully sequential Markov decision processes. It is formulated for settings with no environment interaction and no true reward during training, where the available supervision consists of mixed-quality demonstrations DU=DEDID^U = D^E \cup D^I and a human feedback function h:S×A{1,0,1}h : \mathcal{S}\times\mathcal{A} \to \{-1,0,1\} that labels state–action pairs as good, bad, or neutral. Its defining mechanism is a feedback-tempered surrogate policy πτ\pi_\tau, constructed by modifying action probabilities with per-action temperatures, together with a reverse-KL-based regularizer that encourages the learned policy πθ\pi_\theta to match that surrogate in a single offline training stage (Poole et al., 8 Jul 2026).

1. Problem formulation and alignment objective

FMR is defined in a modified MDP with feedback,

M=(S,A,p,r,h,γ),\mathcal{M} = (\mathcal{S}, \mathcal{A}, p, r, h, \gamma),

but its training regime is explicitly offline imitation: the reward rr is not used, and learning proceeds only from demonstrations and evaluative feedback. The demonstration corpus contains aligned expert trajectories DED^E, imperfect trajectories DID^I, and their union DU=DEDID^U = D^E \cup D^I. The target of alignment is a latent human-aligned policy πh\pi_h, while the learned policy is h:S×A{1,0,1}h : \mathcal{S}\times\mathcal{A} \to \{-1,0,1\}0. A standard coverage assumption is imposed: for any h:S×A{1,0,1}h : \mathcal{S}\times\mathcal{A} \to \{-1,0,1\}1, if h:S×A{1,0,1}h : \mathcal{S}\times\mathcal{A} \to \{-1,0,1\}2 then h:S×A{1,0,1}h : \mathcal{S}\times\mathcal{A} \to \{-1,0,1\}3. This ensures that aligned actions remain in the support of the learned policy (Poole et al., 8 Jul 2026).

Within this formulation, feedback is absolute and local rather than comparative. The signal h:S×A{1,0,1}h : \mathcal{S}\times\mathcal{A} \to \{-1,0,1\}4 designates a specific state–action pair as desirable, undesirable, or neutral. This differs from preference-learning setups that only express relative judgments between trajectories or segments. FMR therefore treats demonstrations as coverage over feasible behavior and feedback as local discrimination over which parts of that behavior should be amplified or suppressed. The result is a single-stage offline alignment objective in which demonstrations and feedback are optimized jointly rather than through separate pretraining, reward-modeling, and reinforcement-learning phases (Poole et al., 8 Jul 2026).

A central misconception addressed by this formulation is that evaluative feedback must be converted into reward. FMR does not create or modify rewards; instead, it reshapes the action distribution directly. This places the method closer to direct policy regularization than to offline RL with learned reward surrogates.

2. Temperature-based manipulation of the policy distribution

The core device in FMR is a per-action temperature h:S×A{1,0,1}h : \mathcal{S}\times\mathcal{A} \to \{-1,0,1\}5, abbreviated h:S×A{1,0,1}h : \mathcal{S}\times\mathcal{A} \to \{-1,0,1\}6, derived from evaluative feedback. Positive feedback corresponds to h:S×A{1,0,1}h : \mathcal{S}\times\mathcal{A} \to \{-1,0,1\}7 in effect, encouraging the action by sharpening the distribution; negative feedback corresponds to h:S×A{1,0,1}h : \mathcal{S}\times\mathcal{A} \to \{-1,0,1\}8, discouraging the action by flattening the distribution; neutral feedback leaves the action unchanged with h:S×A{1,0,1}h : \mathcal{S}\times\mathcal{A} \to \{-1,0,1\}9. The discrete-action mapping is

πτ\pi_\tau0

with hyperparameter πτ\pi_\tau1. For negative feedback on the chosen action, the selected action receives higher temperature and is discouraged. For positive feedback, the formulation raises the temperature of the other actions instead, which is equivalent in a renormalized sense to lowering the temperature of the selected action while avoiding numerical issues associated with πτ\pi_\tau2 (Poole et al., 8 Jul 2026).

Given πτ\pi_\tau3 and πτ\pi_\tau4, FMR defines a feedback-tempered surrogate policy

πτ\pi_\tau5

This construction operates directly in probability space rather than on logits. Actions with lower temperature are up-weighted, and actions with higher temperature are down-weighted. Conceptually, πτ\pi_\tau6 serves as an approximation to the latent aligned policy πτ\pi_\tau7, obtained by correcting the current policy with feedback (Poole et al., 8 Jul 2026).

Alignment is then enforced through the reverse KL

πτ\pi_\tau8

For a fixed state πτ\pi_\tau9, the decomposition

πθ\pi_\theta0

yields the regularization term

πθ\pi_\theta1

For a base imitation-learning objective πθ\pi_\theta2, FMR forms

πθ\pi_\theta3

with πθ\pi_\theta4 controlling alignment strength (Poole et al., 8 Jul 2026).

This formulation admits an entropy interpretation. If πθ\pi_\theta5, then πθ\pi_\theta6, so the regularizer reduces to negative entropy. More generally, FMR can be interpreted as human-controlled entropy regularization: negative feedback increases entropy around undesirable actions, whereas positive feedback decreases entropy around desirable actions (Poole et al., 8 Jul 2026).

3. Integration with imitation learning and offline data

FMR is algorithm-agnostic. It requires only a policy that induces a distribution πθ\pi_\theta7, and can therefore be attached to Behavior Cloning, IQ-Learn, DemoDICE, ReCOIL, and related imitation-learning or Learning from Noisy Demonstrations methods. In actor-only settings, the regularizer is applied directly to the policy. In critic-only settings, Q-values are converted into a distribution, for example by softmax over actions, and FMR is applied to that distribution. In actor–critic settings, it can be applied to both actor and critic distributions (Poole et al., 8 Jul 2026).

For Behavior Cloning, the base loss is

πθ\pi_\theta8

and the combined objective is

πθ\pi_\theta9

For DemoDICE, ReCOIL, and IQ-Learn, the same additive regularization principle is retained, even though the underlying base losses differ in whether they are framed as occupancy matching, actor–critic learning, or soft-Q optimization (Poole et al., 8 Jul 2026).

The interaction between demonstrations and feedback is structurally important. Demonstrations determine which states and actions are available to learn from, including aligned fragments within otherwise imperfect trajectories. Feedback then manipulates the policy’s uncertainty locally. For state–action pairs in M=(S,A,p,r,h,γ),\mathcal{M} = (\mathcal{S}, \mathcal{A}, p, r, h, \gamma),0, negative labels increase temperature and reallocate probability mass away from the labeled action, whereas positive labels reduce relative temperature and concentrate probability on the labeled action. FMR therefore re-weights the policy’s action distribution rather than reweighting the dataset itself. It does not edit labels or trajectories; its effect is realized through gradient updates to the policy (Poole et al., 8 Jul 2026).

The data-collection protocol also matters. In navigation environments, feedback is provided by humans through a replay tool and aligned to states with a 600 ms credit assignment window. In velocity-limited locomotion, the feedback is an oracle proxy derived from cost signals and is mostly negative only. A plausible implication is that the formalism is agnostic to the origin of the evaluative signal so long as it can be represented as offline labels over state–action pairs.

4. Experimental evaluation in Safety Gymnasium

FMR is evaluated in adapted Safety Gymnasium environments designed to separate aptitude from alignment. Rewards are not used for training. Alignment is measured by a misalignment score,

M=(S,A,p,r,h,γ),\mathcal{M} = (\mathcal{S}, \mathcal{A}, p, r, h, \gamma),1

where M=(S,A,p,r,h,γ),\mathcal{M} = (\mathcal{S}, \mathcal{A}, p, r, h, \gamma),2 is the environment cost, and aptitude is measured independently: success rate for navigation and normalized return relative to M=(S,A,p,r,h,γ),\mathcal{M} = (\mathcal{S}, \mathcal{A}, p, r, h, \gamma),3 for velocity tasks (Poole et al., 8 Jul 2026).

The navigation benchmark uses a red spherical agent with lidar observations that must reach a green goal while avoiding unobservable hazards, including floor hazards and hazard walls. Two aligned policies are defined. PathM follows a diagonal corridor through the center and has high overlap with imperfect demonstrations. PathBB follows the right boundary and then the bottom boundary into the goal and has low overlap. The locomotion benchmark discretizes the action space of Swimmer, Hopper, and Walker2D for demonstration collection and defines alignment through maximum M=(S,A,p,r,h,γ),\mathcal{M} = (\mathcal{S}, \mathcal{A}, p, r, h, \gamma),4-velocity constraints: SlowSwim with M=(S,A,p,r,h,γ),\mathcal{M} = (\mathcal{S}, \mathcal{A}, p, r, h, \gamma),5, SlowHop with M=(S,A,p,r,h,γ),\mathcal{M} = (\mathcal{S}, \mathcal{A}, p, r, h, \gamma),6, and SlowWalk with M=(S,A,p,r,h,γ),\mathcal{M} = (\mathcal{S}, \mathcal{A}, p, r, h, \gamma),7 (Poole et al., 8 Jul 2026).

Across Behavior Cloning, IQ-Learn, DemoDICE, and ReCOIL, baseline imitation-learning methods exhibit substantial misalignment, especially when expert data are scarce and imperfect demonstrations dominate. Adding FMR consistently reduces misalignment, often while preserving or improving aptitude. On PathM, baseline misalignment is approximately M=(S,A,p,r,h,γ),\mathcal{M} = (\mathcal{S}, \mathcal{A}, p, r, h, \gamma),8–M=(S,A,p,r,h,γ),\mathcal{M} = (\mathcal{S}, \mathcal{A}, p, r, h, \gamma),9 for BC and IQL and about rr0 for ReCOIL at the 10–50 ratio, whereas ReCOIL+FMR reduces misalignment to around rr1–rr2, corresponding to up to rr3 reduction. On PathBB, where overlap is low, ReCOIL+FMR achieves up to rr4 reduction in misalignment while improving success rates. On SlowSwim, FMR variants retain return near rr5 while cutting misalignment by up to rr6, and ReCOIL+FMR produces misalignment effectively near zero at all ratios. On SlowHop, FMR reduces misalignment by roughly rr7 across algorithms and data ratios while maintaining competitive returns. On SlowWalk, with positive feedback on rr8 and rr9, ReCOIL+FMR attains the best return–alignment tradeoff and up to DED^E0 reduction in misalignment (Poole et al., 8 Jul 2026).

The ablation results clarify how the method uses feedback. Treating feedback as reward through DVL often preserves high return but leaves misalignment high or worsens it in low-expert settings. CPL, which uses feedback-summed segment scores in a preference-learning style, performs reasonably on some tasks but degrades in more complex or high-dimensional action spaces such as SlowWalk and still underperforms ReCOIL+FMR in alignment. Experiments with only negative feedback show little degradation on most tasks, indicating that FMR can operate effectively with negative-only supervision. Feedback sparsity experiments, in which only DED^E1 or DED^E2 of trajectories carry feedback, show graceful degradation and usually remain superior to baselines without FMR, especially when the retained feedback is targeted rather than randomly thinned (Poole et al., 8 Jul 2026).

5. Relation to adjacent methods and conceptual lineage

FMR occupies a distinct position relative to standard RLHF, learning-from-preferences, and evaluative-feedback RL methods. In prevailing RLHF pipelines for LLMs, demonstrations and human judgments are typically separated into multiple stages: supervised learning on demonstrations, reward-model fitting from pairwise preferences, and RL fine-tuning against the learned reward. Those methods are usually framed as contextual bandits rather than fully sequential MDPs. By contrast, FMR performs single-stage, fully offline integration of demonstrations and feedback in a sequential setting, and the feedback signal is evaluative and absolute rather than pairwise and relative (Poole et al., 8 Jul 2026).

This distinction has methodological consequences. Pairwise preference methods only determine which of two segments is better; they do not establish that either segment is aligned. FMR instead provides local, state–action-level control over where probability mass should contract or disperse. Online evaluative-feedback methods such as TAMER and COACH use human feedback as pseudo-reward with credit assignment, but they require online interaction and do not formulate feedback as a direct distribution-shaping regularizer in offline imitation learning. FMR therefore should not be understood as offline reward learning in disguise; its objective acts directly on the policy distribution (Poole et al., 8 Jul 2026).

A related but terminologically distinct use of feedback manipulation appears in recommender systems. “DPR: An Algorithm Mitigate Bias Accumulation in Recommendation feedback loops” does not use the label FMR, but it describes an analogous strategy in which training feedback is explicitly manipulated, re-weighted, or regularized so that the model does not inherit and amplify exposure bias under feedback loops. In that setting, Dynamic Personalized Ranking acts as a feedback re-weighting or regularization scheme, and Universal Anti-False Negative acts as an additional regularization term targeted at false negatives in implicit feedback (Xu et al., 2023). This suggests a broader abstraction of “feedback manipulation regularization” as direct correction of feedback-distorted learning signals, even though the formal objects differ sharply between recommendation and sequential policy learning.

6. Practical considerations, limitations, and prospective extensions

Several implementation details are emphasized. A default choice of DED^E3 works well across tasks, while DED^E4 is a good default regularization strength; DED^E5 is reported to help in Walker2D, where the action space is large. Larger DED^E6 can make temperatures too extreme, and very large DED^E7 can over-regularize and hurt aptitude. In navigation, a 600 ms credit assignment window is used by default and has only modest sensitivity. When expert demonstrations are scarce, oversampling DED^E8 to roughly match the size of DED^E9 stabilizes training across algorithms. In actor–critic settings such as DemoDICE+FMR and ReCOIL+FMR, applying FMR to both actor and critic distributions is reported as beneficial (Poole et al., 8 Jul 2026).

The main limitation is dependence on overlap and feedback coverage. FMR is most effective when imperfect demonstrations contain aligned sub-trajectories that feedback can identify and amplify. When DID^I0 is almost entirely misaligned or irrelevant, the method has little aligned behavior to exploit. Manual evaluative feedback also remains expensive, even though targeted and partial annotation can still be effective. Experimental validation is currently limited to discrete action spaces; a generalized continuous-action variant,

DID^I1

is proposed and behaves similarly in discrete environments, but high-dimensional continuous control remains unvalidated. The framework also lacks formal convergence or regret bounds, and its notion of alignment is environment-specific, being defined through safety costs in Safety Gymnasium (Poole et al., 8 Jul 2026).

The broader implications are nevertheless clear within the stated scope. FMR demonstrates that offline data and evaluative feedback can substantially align policies without online RL fine-tuning, including in limited-data regimes with scarce aligned demonstrations and noisy mixed-quality data. Potential extensions identified for LLMs, robotics, autonomous driving, and healthcare decision support all retain the same conceptual move: using feedback as a direct distribution-shaping regularizer rather than as a learned reward. This suggests that FMR is best understood as a general recipe for human-guided local entropy modulation in sequential decision-making, with current evidence concentrated in safety-oriented offline imitation learning (Poole et al., 8 Jul 2026).

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 Feedback Manipulation Regularization (FMR).