Mental Reward Models
- Mental Reward Models are frameworks that incorporate internal cognitive signals (e.g., latent pain, metacognitive states) with external outcomes to guide learning.
- They employ diverse methodologies such as latent-state filtering, metacognitive trajectory supervision, and appraisal-based reward shaping to enhance process quality.
- Applications span reinforcement learning, social agent modeling, and therapeutic dialogue, where aligning internal evaluations with external metrics improves performance.
A mental reward model is a reward construction in which optimization is guided not only by externally observed task outcomes but also by inferred internal states, metacognitive process variables, appraisal checks, social belief estimates, or latent satisfaction. In contemporary arXiv literature, the term covers several technically distinct families: subjective well-being functions that subtract inferred pain-belief from reward, metacognitive trajectory rewards for LLM reasoning, appraisal-based affective rewards derived from temporal-difference signals, intrinsic rewards for predicting other agents’ beliefs, distributional mental rewards for recommendation, and mechanistic reward-valuation signals in multimodal models (Petrowski et al., 6 Jan 2026, Chen et al., 22 May 2026, Zhang et al., 2023, Oguntola et al., 2023, Zhao et al., 26 Sep 2025, Honarmand et al., 7 Jul 2026). The unifying feature is that reward is treated as a model of mental or internally evaluated significance rather than as a direct synonym for environmental payoff.
1. General form of mental reward models
Across the literature, mental reward models are defined by augmenting or replacing an objective reward with variables intended to encode internal evaluation. In introspective reinforcement learning, the subjective reward is
where the final term is a filtered posterior over a latent pain state. In Theory-of-Mind multi-agent RL, the total reward is
In appraisal-based modeling, the mental reward is an additive combination of novelty, goal relevance, goal conduciveness, and power. In metacognitive LLM training, the trajectory reward is
In sequential recommendation, the reward is explicitly modeled as a distribution, , rather than as a deterministic scalar (Petrowski et al., 6 Jan 2026, Oguntola et al., 2023, Zhang et al., 2023, Chen et al., 22 May 2026, Zhao et al., 26 Sep 2025).
Taken together, these formulations separate at least four axes of design. First, the internal signal may be latent-state belief, as in hidden Markov pain inference, or an explicit graded evaluation, as in metacognitive knowledge coverage and regulation fidelity. Second, the reward may be step-level, trajectory-level, or both. Third, the mental variable may be self-directed, other-directed, or socially comparative. Fourth, the mental reward may serve either as the principal learning target or as an auxiliary alignment signal added to an existing objective. This suggests that “mental reward model” is best understood as a family resemblance term for psychologically structured reward design, rather than a single canonical architecture.
2. Introspective reward and internal-state inference
In “Exploration Through Introspection: A Self-Aware Reward Model,” the core mental variable is a latent two-state HMM with and observations . The model performs online filtering with
and uses the posterior pain-belief as an aversive term inside a subjective well-being function. The framework includes normal and chronic pain parameterizations from Eckert et al. (2022): the chronic model has sticky transitions and ambiguous emissions relative to the normal model, and the initial distribution is also biased toward pain. Q-learning is then performed on the subjective reward rather than on the bare environmental reward (Petrowski et al., 6 Jan 2026).
The experiments place agents in a stationary grid with a single “food” cell and in a non-stationary variant where the food jumps to a different corner every steps. The reported metrics are Cumulative Objective Reward, momentary and cumulative well-being, and pain-belief trajectories. In almost all reward categories, normal- and chronic-pain agents significantly outperformed their “no-pain” baselines under paired 0-tests with 1. In the stationary setting, normal and chronic models achieved 2 versus baseline 3. In the non-stationary setting, the chronic model sometimes exceeded the normal model, for example in Objective+Expect with chronic 4 versus normal 5, but at the cost of a persistently negative cumulative well-being. The chronic agent’s relief-seeking cycle is described as paralleling addictive-like behaviour (Petrowski et al., 6 Jan 2026).
A related but more abstract treatment of distorted reward processing appears in Human-Based Thompson Sampling. There the reward model is not an inferred latent state but a parameterized update bias in a Beta-Bernoulli bandit:
6
The four parameters 7 encode retention of past positive evidence, sensitivity to current positive reward, retention of past negative evidence, and sensitivity to current negative reward. Disorder-inspired settings include Addiction, ADHD, Alzheimer’s, Chronic Pain, bvFTD, and Parkinson’s; Chronic Pain is parameterized as 8, i.e. down-weighted immediate positive reward with full retention of negative history. In all 12 evaluated settings, at least one HBTS variant strictly outperformed standard Thompson Sampling (Bouneffouf et al., 2017).
3. Metacognition and appraisal as reward
In “Metacognition as Reward: Reinforcing LLM Reasoning via Knowledge and Regulation Signals,” the rollout is explicitly scaffolded into Metacognitive Knowledge, Metacognitive Regulation, optional LOOKBACK, and final answer. A grader returns the number of covered knowledge units 9, recovered units 0, plan-alignment score 1, shortcut flag 2, and correctness 3. The two process rewards are
4
with 5, and the full trajectory reward is
6
The default setting uses 7, yielding 8. Optimization is performed with group-relative advantages and the clipped DAPO objective. On 22 benchmarks, MaR improved performance by up to 9 over the base model and up to 0 over vanilla DAPO; on out-of-domain long-context tasks it increased 1 by 2, 3 by 4, and 5 by 6. Ablations showed that removing either metacognitive component hurt performance more than removing final correctness, indicating that the process-level supervision is not reducible to outcome-only reward (Chen et al., 22 May 2026).
A different formal route is taken by Zhang, Broekens, and Jokinen, who derive mental reward from appraisal theory and temporal-difference learning. They define novelty as
7
goal relevance as
8
goal conduciveness by clipping the TD error and rescaling it to 9,
0
and power as the mean-to-minimum action-value gap in the successor state. The unified mental reward is then
1
with identity mappings for each appraisal component. The same appraisal vector is also sent to a pre-trained SVM for discrete emotion classification. In the “Fear” vignette, the appraisal vector 2 yields “Fear” as the modal emotion. The authors explicitly note that substituting 3 into the Q-update is proposed, but not empirically evaluated beyond the vignette studies (Zhang et al., 2023).
These two strands share a structural principle: reward is moved upstream from final correctness or external payoff to intermediate evaluative variables that are intended to reflect how the agent monitors or appraises its own cognition. A plausible implication is that mental reward models are especially attractive where sparse terminal rewards fail to constrain process quality.
4. Social and comparative mental reward
In social reinforcement learning, Oguntola et al. define a mental reward from second-order belief prediction. Each agent maintains a first-order belief vector 4 and a second-order belief matrix 5, where 6 is agent 7’s prediction of agent 8’s first-order belief. The intrinsic reward is the negative average prediction error,
9
with 0 equal to MSE or CE depending on whether beliefs are continuous or discrete. This reward is added to the task reward and optimized under PPO within a belief-grounded policy architecture that separates a supervised belief predictor from a residual representation and a separate ToM predictor. In Particle World Physical Deception, equipping the good team with the ToM intrinsic reward increased mean cumulative episode reward from 1 for “No belief modules” to 2 for “2nd-order ToM (good agents),” while the adversary reward dropped from 3 to 4 in that setting. The paper characterizes this as an approximately 5 boost for the good team over the no-belief baseline (Oguntola et al., 2023).
A contrasting social formulation appears in the constructive study of social comparison without explicit inference of others’ reward values. Three probabilistic-generative models are compared: the Internal Prediction Model (IPM), which infers the partner’s subjective values; the No Comparison Model (NCM), which ignores partner information; and the External Comparison Model (ECM), which incorporates partner rewards directly at the top situation-awareness node. All are implemented as multi-layered, multimodal latent Dirichlet allocation variants with Gibbs sampling and message passing. Evaluation uses the Rand Index for classification of subjective values across six experimental conditions. The reported scores are 6 for NCM, 7 for IPM, and 8 for ECM, with chance level approximately 9. Under these settings, the result suggests that social comparison relies on objective reward differences rather than on inferences about subjective states (Taniuchi et al., 21 Dec 2025).
The juxtaposition is instructive. In one line of work, modeling others’ beliefs is itself rewarded. In the other, the best-fitting social valuation model omits explicit partner-value inference. This indicates that “mental reward” in social settings can refer either to explicit Theory-of-Mind competence or to a more parsimonious constructive use of social information.
5. Alignment in counseling dialogue, therapeutic reward models, and recommendation
In mental-health counseling dialogue generation, READER formulates the action as a structured pair 0: first a therapist dialogue-act, then the utterance. Its reward blends ROUGE-1 1, BERTScore, an act-compliance score 2 from a pre-trained SPARTA classifier, and a KL regularizer against a frozen GPT-2 reference policy:
3
After grid search, the coefficients are 4, 5, 6, and 7. PPO uses 8, 9, and GAE-0. Ablations show that removing the ROUGE term drops ROUGE-1 1 from 2 to 3 and METEOR from 4 to 5, while removing the act-compliance term reduces ROUGE-2 recall from 6 to 7 and BERTScore from 8 to 9. The full reward combination yields the best reported gains on HOPE (Srivastava et al., 2023).
Reward modeling for therapy is also treated at the benchmark level in PRMB, a benchmark for long-horizon CBT-based counseling dialogue. PRMB spans 6 sessions and 21 negative scenarios grouped into four clusters, uses progressive summarization to preserve cross-session state, and contains 0 real or synthetic CBT cases, 1 prompts, 2 pairwise preference pairs, and 3 Best-of-4 queries. Evaluation uses pairwise accuracy, Best-of-4 accuracy, and Spearman’s rank correlation between benchmark ranking and downstream Best-of-5 BERTScore ranking. The reported correlation is 6 with 7, compared with 8 and 9 for RewardBench2. PRMB also reveals large pairwise-to-BoN gaps of 0–1 percentage points, session-specific degradation in Sessions 3–5, and the weakest BoN performance on subtle cumulative harms in Cluster IV at approximately 2 (Zhou et al., 12 Mar 2026).
In recommendation, MTRec treats user behavior as a user-centric MDP 3, where the reward is a random mental reward rather than a deterministic click-derived scalar. The model learns a distributional reward with quantile inverse RL, recovers the mean reward as
4
and then uses 5 as auxiliary supervision for downstream recommenders. Offline results on Amazon Books and Electronics show AUC gains of approximately 6–7 and NCIS gains of approximately 8–9. In Virtual Taobao, PPO improves episodic CTR from 00 to 01 and SAC from 02 to 03 when mental reward is added. In an industrial short-video platform, the deployed system increases average user viewing time by 04 over 7 days (Zhao et al., 26 Sep 2025).
These applications differ in domain, but they share the same correctional aim: the learned or engineered reward is intended to align training with latent quality criteria that raw interaction signals do not fully capture.
6. Mechanistic reward valuation, active mental imagery, and scope
Honarmand et al. give the most explicitly mechanistic account of a mental reward model. Adapting the Monetary Incentive Delay task to text for a VLM, they define a neuron-level reward signal
05
select units satisfying 06, and identify roughly 07 of the model’s neurons as NAc-selective, all in late MLP layers 08–09. Cross-frame validation over four paraphrased MID prompts gives a correlation 10 in unit identity. During incentivized trials, the reward signal is causally perturbed by activation patching, 11, or by gain scaling 12. The resulting behavioral changes mirror anhedonia: in ASDiv-EEfRT, mean points fall from approximately 13 to approximately 14, a 15 drop with 16, while forced-choice control accuracy remains 17 with 18. In Probability-EEfRT, 19 collapses by 20 overall with 21, yet both intact and perturbed models compute 22 on 23 of trials. Psychometric analogues shift in the expected direction: DARS decreases by 24, MAP-SR by 25, and AES increases by 26, all with 27 (Honarmand et al., 7 Jul 2026).
A separate expansion of the concept appears in SpatialDreamer, where active mental imagery is trained by Geometric Policy Optimization. GeoPO replaces episode-only reward with tree-structured sampling, bottom-up step-level rewards,
28
and a combined reward
29
with geometric penalties for redundant or conflicting actions and 30. The method is designed for long-horizon spatial reasoning with imagined views from a world model. Reported ablations attribute a 31 drop on MindCube-Tiny to removing the geometric penalty, and the full model improves over GRPO on several benchmarks, including 32 versus 33 on MindCube-Tiny and 34 on SAT-Real/Synthesized versus approximately 35 for GRPO (Cao et al., 8 Dec 2025).
A common misconception is that a mental reward model is necessarily a reward model for mental-health applications. The current literature is broader. It includes models of pain-belief, metacognitive monitoring, appraisal, social comparison, reward valuation circuitry, recommendation satisfaction, and active mental imagery. Another misconception is that such models merely add heuristic reward shaping. The cited work instead shows multiple formal commitments: latent-state filtering, structured trajectory grading, probabilistic generative modeling, inverse reinforcement learning, causal activation patching, and dense credit assignment. This suggests that the central research question is not whether reward should be “mental,” but which internal variables can be formalized reliably enough to improve adaptation, alignment, or interpretability without collapsing task competence.