跳到论文正文
W Papers独立论文阅读器 arXiv:2609.39561 论文正文 计算阅读时间 · 0%
引用这篇论文根据当前论文元数据生成,正式投稿前建议核对。

    
  
Skip to main content
Why HTML? Back to Abstract Download PDF
License: CC BY-NC-ND 4.0
arXiv:2609.39561v1 [cs.LG] 30 Sep 2026

Candidate Retention for Abductive Learning

Hao-Yuan He Affiliation: School of Artificial Intelligence, Nanjing University    Yu Liu Affiliation: National Key Laboratory for Novel Software Technology, Nanjing University    Ming Li Affiliation: {hehy, liuy, lim}@lamda.nju.edu.cn
Abstract

Abductive learning combines neural perception with symbolic reasoning, using explanations generated by abduction to supervise the perception model. Multiple valid explanations of the same symbolic target can assign conflicting labels to the same inputs. Common policies select a single candidate as a pseudo-label, which may reinforce mistaken assignments, or weight all candidates, which may spread supervision across competing labels. These risks motivate selecting a retained subset to balance supervision sharpness and model-mass coverage. To guide this choice, we bound the coordinate-level supervision error using retained uncertainty, discarded model mass, and model mismatch. For a fixed model and training pair, only the first two terms depend on the retained set. We propose Abductive Candidate Retention (ACR), which uses these terms to guide greedy additions, accepting a candidate when its recovered mass exceeds the increase in retained uncertainty. Experiments show that ACR improves concept accuracy over single-candidate baselines and A3​BL\text{A}^{3}\text{BL} in most evaluated aggregated mod-addition settings. Objective ablations support the joint use of uncertainty and posterior mass.

1 Introduction

Neuro-symbolic learning integrates neural perception with symbolic reasoning to learn from data and background knowledge. Abductive learning (ABL, Zhou, 2019) combines neural perception with symbolic reasoning through abduction (Magnani,, 2009): a perception model proposes latent concepts from raw inputs, and when these proposals are inconsistent with the observed evidence, abduction generates candidate concept assignments that entail it. These candidates then supervise the perception model.

However, the background knowledge does not always determine a unique candidate. Consider 𝚂𝚄𝙼⁡([Uncaptioned image],[Uncaptioned image])=𝟹\mathtt{SUM({\raisebox{-2.39996pt}{\includegraphics[height=10.79993pt,width=10.79993pt,keepaspectratio]{figures/example_digit_1.png}}},{\raisebox{-2.39996pt}{\includegraphics[height=10.79993pt,width=10.79993pt,keepaspectratio]{figures/example_digit_2.png}}})=3}: the addition rule admits [[Uncaptioned image]=𝟶,[Uncaptioned image]=𝟹]\mathtt{[{\raisebox{-2.39996pt}{\includegraphics[height=10.79993pt,width=10.79993pt,keepaspectratio]{figures/example_digit_1.png}}}=0,{\raisebox{-2.39996pt}{\includegraphics[height=10.79993pt,width=10.79993pt,keepaspectratio]{figures/example_digit_2.png}}}=3]}, [[Uncaptioned image]=𝟹,[Uncaptioned image]=𝟶]\mathtt{[{\raisebox{-2.39996pt}{\includegraphics[height=10.79993pt,width=10.79993pt,keepaspectratio]{figures/example_digit_1.png}}}=3,{\raisebox{-2.39996pt}{\includegraphics[height=10.79993pt,width=10.79993pt,keepaspectratio]{figures/example_digit_2.png}}}=0]}, and other assignments. Because different candidates label the same images differently, they create ambiguity in the concept-level supervision.

Figure 1: Candidate retention in the ABL loop. Upper: abduction returns 𝒔⁡(𝒙,y){\bm{s}}({\bm{x}},y); supervision construction marginalizes the retained candidates’ weighted labels to train the perception model. Lower: three retention policies with schematic candidate weights p~\tilde{p}. Shading indicates which candidates are retained.

Classical ABL selects one candidate as a pseudo-label using the current model (Dai et al.,, 2019). A3​BL\text{A}^{3}\text{BL} normalizes all valid candidates into a posterior and marginalizes it into soft labels (He et al., 2024b, ). Selecting one explanation yields sharp labels but can reinforce a mistaken assignment. Retaining more explanations preserves alternatives, while their disagreements can spread supervision across competing labels. We call the choice of which candidates to include candidate retention.

A candidate’s contribution to supervision depends on the label support already retained. Two candidates with equal weights can therefore change supervision differently: one may reinforce supported labels, while the other adds competing labels. This combination-dependent effect motivates evaluating each addition through the supervision formed by the retained set.

We propose Abductive Candidate Retention (ACR), which evaluates additions through their recovered model mass and change in coordinate uncertainty (Figure 1). An upper bound on coordinate supervision error contains retained uncertainty, discarded model mass, and model mismatch. For a fixed model and training pair, only the first two depend on the retained set. Let Mθ​(R,𝒙,y)M_{\theta}(R;{\bm{x}},y) denote the retained mass and Uθ​(R,𝒙,y)\mathrm{U}_{\theta}(R;{\bm{x}},y) the average coordinate uncertainty. We score a retained set RR by

Jθ​(R,𝒙,y)=Uθ​(R,𝒙,y)+1−Mθ​(R,𝒙,y).J_{\theta}(R;{\bm{x}},y)=\mathrm{U}_{\theta}(R;{\bm{x}},y)+1-M_{\theta}(R;{\bm{x}},y).

ACR starts from the highest-weight candidate and adds the candidate that most decreases this criterion, stopping when no addition improves it. Marginalizing the retained weights then supplies concept targets for the neural update.

Our contributions:

  1. 1.

    We analyze candidate retention through the concept supervision induced by a subset and give an upper bound on its coordinate-level error.

  2. 2.

    We develop a greedy rule that evaluates added candidate mass together with changes in coordinate uncertainty, adapting the retained set’s composition and size.

  3. 3.

    We show concept-accuracy gains on most evaluated mod-addition settings, with controlled comparisons and training diagnostics, and evaluate ACR on chess attack and judicial sentencing.

2 Problem Setup

Let 𝒳{\cal X} be the input space, 𝒵{\cal Z} a finite concept set, and 𝒴{\cal Y} the target space. In typical ABL tasks, a single symbolic target constrains multiple inputs jointly. Each training example therefore contains an input sequence 𝒙=(xj)j=1m∈𝒳m{\bm{x}}=(x_{j})_{j=1}^{m}\in{\cal X}^{m} and an observed target y∈𝒴y\in{\cal Y}. The concept zj∈𝒵z_{j}\in{\cal Z} for each input xjx_{j} is hidden.

ABL (Zhou,, 2019) combines a perception model with background knowledge KB. For each input xx, the model assigns a probability fθ​(k∣x)f_{\theta}(k\mid x) to every concept k∈𝒵k\in{\cal Z}.

The knowledge base defines the candidates that entail the observed target:

𝒔⁡(𝒙,y)={𝒄∈𝒵m:𝒄∧KB⊧y}.{\bm{s}}({\bm{x}},y)=\{{\bm{c}}\in{\cal Z}^{m}:{\bm{c}}\land\text{KB}\models y\}.

Training reveals yy, but it does not identify which 𝒄∈𝒔⁡(𝒙,y){\bm{c}}\in{\bm{s}}({\bm{x}},y) matches the inputs. ABL uses these valid candidates to learn the hidden concepts.

Example 1.

Consider SUM([Uncaptioned image],[Uncaptioned image])=3. The input contains two unlabeled digit images, and 3 is their observed sum. The valid candidates are [0,3], [1,2], [2,1], and [3,0]. Only one describes the image contents, but its digit labels are unavailable during training.

2.1 Existing ABL Supervision

Classical ABL selects one valid candidate and uses it as supervision (Dai et al.,, 2019). It commonly ranks candidates by confidence or Hamming distance. Under the factorized perception model, candidate confidence is

pθ​(𝒄∣𝒙)=∏j=1mfθ​(cj∣xj).p_{\theta}({\bm{c}}\mid{\bm{x}})=\prod_{j=1}^{m}f_{\theta}(c_{j}\mid x_{j}).

A Hamming score measures the negative distance between 𝒄{\bm{c}} and the predicted concept sequence. Writing either score generically as scoreθ\operatorname{score}_{\theta}, ABL selects

𝒄^θ∈arg⁡max𝒄∈𝒔⁡(𝒙,y)​scoreθ⁡(𝒄;𝒙).\hat{{\bm{c}}}_{\theta}\in\arg\max_{{\bm{c}}\in{\bm{s}}({\bm{x}},y)}\operatorname{score}_{\theta}({\bm{c}};{\bm{x}}).

The selected candidate provides hard labels for updating fθf_{\theta}. The updated model then reranks the candidates.

A3​BL\text{A}^{3}\text{BL} keeps all valid candidates and weights them by confidence (He et al., 2024b, ). Their total confidence is

Zθ​(𝒙,y)=∑𝒄∈𝒔⁡(𝒙,y)pθ​(𝒄∣𝒙).Z_{\theta}({\bm{x}},y)=\sum_{{\bm{c}}\in{\bm{s}}({\bm{x}},y)}p_{\theta}({\bm{c}}\mid{\bm{x}}).

Normalizing by Zθ​(𝒙,y)Z_{\theta}({\bm{x}},y) gives the candidate posterior

wθ​(𝒄∣𝒙,y)=pθ​(𝒄∣𝒙)Zθ​(𝒙,y),𝒄∈𝒔⁡(𝒙,y).w_{\theta}({\bm{c}}\mid{\bm{x}},y)=\frac{p_{\theta}({\bm{c}}\mid{\bm{x}})}{Z_{\theta}({\bm{x}},y)},\qquad{\bm{c}}\in{\bm{s}}({\bm{x}},y).

We assume 𝒔⁡(𝒙,y)≠∅{\bm{s}}({\bm{x}},y)\neq\varnothing and Zθ​(𝒙,y)>0Z_{\theta}({\bm{x}},y)>0.

A3​BL\text{A}^{3}\text{BL} marginalizes this posterior at each sequence position to construct soft concept targets for fθf_{\theta}.

Classical ABL and A3​BL\text{A}^{3}\text{BL} provide the single-candidate and posterior-marginalization reference policies. We next consider how the retained subset shapes concept supervision and use this information to guide selection.

3 Abductive Candidate Retention

ACR scores retained sets, converts them into concept targets, and constructs them by greedy selection. Throughout this section, wθ​(𝒄∣𝒙,y)w_{\theta}({\bm{c}}\mid{\bm{x}},y) denotes a normalized model-induced distribution over the nonempty valid set 𝒔⁡(𝒙,y){\bm{s}}({\bm{x}},y). The product posterior above is one choice; task-specific scores are given in Appendix B.

3.1 A Retention Criterion

For a training pair, let R⊆𝒔⁡(𝒙,y)R\subseteq{\bm{s}}({\bm{x}},y) be the retained candidate set. Conditioning the candidate distribution on RR gives its retained mass and conditional distribution:

Mθ​(R,𝒙,y)\displaystyle M_{\theta}(R;{\bm{x}},y) =∑𝒄∈Rwθ​(𝒄∣𝒙,y),\displaystyle=\sum_{{\bm{c}}\in R}w_{\theta}({\bm{c}}\mid{\bm{x}},y),
qθ,R​(𝒄∣𝒙,y)\displaystyle q_{\theta,R}({\bm{c}}\mid{\bm{x}},y) =wθ​(𝒄∣𝒙,y)Mθ​(R,𝒙,y)𝕀[𝒄∈R],\displaystyle=\frac{w_{\theta}({\bm{c}}\mid{\bm{x}},y)}{M_{\theta}(R;{\bm{x}},y)}\mathbb{I}[{\bm{c}}\in R],

where Mθ​(R,𝒙,y)>0M_{\theta}(R;{\bm{x}},y)>0.

At position jj, the marginal of qθ,Rq_{\theta,R} is

πθ,R,j(k)=Pr𝒄∼qθ,R(⋅∣𝒙,y)(cj=k).\pi_{\theta,R,j}(k)=\Pr_{{\bm{c}}\sim q_{\theta,R}(\cdot\mid{\bm{x}},y)}(c_{j}=k).

Retained uncertainty averages one minus the largest marginal probability across positions:

Uθ​(R,𝒙,y)=1m​∑j=1m(1−maxk∈𝒵⁡πθ,R,j​(k)).\mathrm{U}_{\theta}(R;{\bm{x}},y)=\frac{1}{m}\sum_{j=1}^{m}\left(1-\max_{k\in{\cal Z}}\pi_{\theta,R,j}(k)\right).

A lower Uθ​(R,𝒙,y)\mathrm{U}_{\theta}(R;{\bm{x}},y) gives sharper concept targets. The discarded posterior mass is 1−Mθ​(R,𝒙,y)1-M_{\theta}(R;{\bm{x}},y).

To relate the retained supervision to concept error, let P⋆(⋅∣𝒙,y)P^{\star}(\cdot\mid{\bm{x}},y) be the unknown true distribution over valid candidates. It is a point mass when each input sequence has one true candidate.

The retained prediction at position jj is

z^R,j​(𝒙,y)∈arg⁡maxk∈𝒵​πθ,R,j​(k).\hat{z}_{R,j}({\bm{x}},y)\in\arg\max_{k\in{\cal Z}}\pi_{\theta,R,j}(k).

Its ideal risk is the expected coordinate error under P⋆P^{\star}:

ℛ⋆(R;𝒙,y)=𝔼𝒄∼P⋆(⋅∣𝒙,y)[1m∑j=1m𝕀[z^R,j(𝒙,y)≠cj]].\mathcal{R}^{\star}(R;{\bm{x}},y)={\mathbb{E}}_{{\bm{c}}\sim P^{\star}(\cdot\mid{\bm{x}},y)}\left[\frac{1}{m}\sum_{j=1}^{m}\mathbb{I}[\hat{z}_{R,j}({\bm{x}},y)\neq c_{j}]\right].

Since P⋆P^{\star} is unknown, we compare the induced supervision with this distribution through the following upper bound.

Theorem 1.

For any positive-mass set R⊆𝐬⁡(𝐱,y)R\subseteq{\bm{s}}({\bm{x}},y),

ℛ⋆​(R,𝒙,y)\displaystyle\mathcal{R}^{\star}(R;{\bm{x}},y) ≤Uθ​(R,𝒙,y)+TV⁡(P⋆,wθ)\displaystyle\leq\mathrm{U}_{\theta}(R;{\bm{x}},y)+\mathrm{TV}\!\left(P^{\star},w_{\theta}\right)
+1−Mθ​(R,𝒙,y),\displaystyle+1-M_{\theta}(R;{\bm{x}},y),

where TV\mathrm{TV} denotes total variation distance and both distributions are conditioned on (𝐱,y)({\bm{x}},y).

Proof.

Write P=P⋆P=P^{\star}, w=wθw=w_{\theta}, qR=qθ,Rq_{R}=q_{\theta,R}, and MR=Mθ​(R)M_{R}=M_{\theta}(R), suppressing (𝒙,y)({\bm{x}},y). The coordinate loss ℓR(𝒄)=m−1∑j𝕀[z^R,j≠cj]\ell_{R}({\bm{c}})=m^{-1}\sum_{j}\mathbb{I}[\hat{z}_{R,j}\neq c_{j}] lies in [0,1][0,1], so

𝔼P​[ℓR]≤𝔼qR​[ℓR]+TV⁡(P,qR).{\mathbb{E}}_{P}[\ell_{R}]\leq{\mathbb{E}}_{q_{R}}[\ell_{R}]+\mathrm{TV}(P,q_{R}).

The first term equals Uθ​(R)\mathrm{U}_{\theta}(R). On RR, qR=w/MR≥wq_{R}=w/M_{R}\geq w; outside RR, qR=0q_{R}=0. The absolute differences sum to 1−MR1-M_{R} on each part, so TV⁡(w,qR)=1−MR\mathrm{TV}(w,q_{R})=1-M_{R}. By the triangle inequality,

TV⁡(P,qR)\displaystyle\mathrm{TV}(P,q_{R}) ≤TV⁡(P,w)+TV⁡(w,qR)\displaystyle\leq\mathrm{TV}(P,w)+\mathrm{TV}(w,q_{R})
=TV⁡(P,w)+1−MR.\displaystyle=\mathrm{TV}(P,w)+1-M_{R}.

Combining the two inequalities proves the result. ∎

Interpretation.

Under qθ,Rq_{\theta,R}, the coordinate error equals retained uncertainty. Model mismatch and discarded mass bound the change in expected error when moving to P⋆P^{\star}. Thus retained mass describes coverage under the model distribution, and its relation to the true explanation depends on model mismatch. For a fixed model and training pair, the mismatch term is constant across RR. ACR uses retained uncertainty and discarded mass, the two subset-dependent terms in the bound, to score candidate additions:

Jθ​(R,𝒙,y)=Uθ​(R,𝒙,y)+1−Mθ​(R,𝒙,y).J_{\theta}(R;{\bm{x}},y)=\mathrm{U}_{\theta}(R;{\bm{x}},y)+1-M_{\theta}(R;{\bm{x}},y).

Let CKC_{K} contain the min⁡(K,|𝒔|)\min(K,|{\bm{s}}|) highest-weight candidates, with fixed tie-breaking. Weights remain normalized over 𝒔{\bm{s}}, so discarded mass includes candidates outside this search pool. The full valid set globally minimizes JθJ_{\theta}; the criterion alone does not favor partial retention. ACR uses it to evaluate individual additions along a greedy path from the highest-weight singleton and stops when no addition improves it. This local rule can yield intermediate subsets. Their benefit over the full set is evaluated empirically and is not guaranteed by the bound.

3.2 Retained Concept Supervision

At round tt, the current parameters θt\theta_{t} and retained set RtR_{t} define soft targets πt,j​(k)=πθt,Rt,j​(k)\pi_{t,j}(k)=\pi_{\theta_{t},R_{t},j}(k) and mass Mt=Mθt​(Rt,𝒙,y)M_{t}=M_{\theta_{t}}(R_{t};{\bm{x}},y). Holding these quantities fixed during the neural update gives the normalized cross-entropy loss

ℒret​(θ,πt)=∑j=1m∑k∈𝒵πt,j​(k)​[−log⁡fθ​(k∣xj)].{\mathcal{L}}_{\mathrm{ret}}(\theta;\pi_{t})=\sum_{j=1}^{m}\sum_{k\in{\cal Z}}\pi_{t,j}(k)[-\log f_{\theta}(k\mid x_{j})].

Marginalization pools label support across joint explanations. Several candidates can agree at one position while assigning different labels elsewhere; their weights add to the same target entry at that position. Every input in a sequence receives a target derived from the same retained distribution. A singleton produces one-hot targets, while retaining the full valid set recovers its full-distribution marginals. Intermediate sets preserve the alternatives that contribute to the selected supervision.

In aggregated mod addition, the implementation uses the original retained weights to form the targets:

at,j​(k)\displaystyle a_{t,j}(k) =∑𝒄∈Rtwθt(𝒄∣𝒙,y)𝕀[cj=k]\displaystyle=\sum_{{\bm{c}}\in R_{t}}w_{\theta_{t}}({\bm{c}}\mid{\bm{x}},y)\mathbb{I}[c_{j}=k]
=Mt​πt,j​(k).\displaystyle=M_{t}\pi_{t,j}(k).

Its loss is Mt​ℒret​(θ,πt)M_{t}{\mathcal{L}}_{\mathrm{ret}}(\theta;\pi_{t}), with MtM_{t} also fixed during the update. The conditional marginals specify the relative label distribution, and retained mass weights the training example. Appendix B gives the scoring and weighting conventions. The standard EM connection for fixed sets, product-posterior weights, and normalized targets is described in Appendix A.1.

3.3 Greedy Candidate Retention

We construct R⊆CKR\subseteq C_{K} by forward selection, starting with the highest-weight candidate. The set expands only when adding a candidate lowers the criterion.

To see how a candidate changes supervision, write M=Mθ​(R,𝒙,y)M=M_{\theta}(R;{\bm{x}},y), w𝒄=wθ​(𝒄∣𝒙,y)w_{{\bm{c}}}=w_{\theta}({\bm{c}}\mid{\bm{x}},y), and πj=πθ,R,j\pi_{j}=\pi_{\theta,R,j}. Adding 𝒄∉R{\bm{c}}\notin R yields marginals πj+=πθ,R∪{𝒄},j\pi_{j}^{+}=\pi_{\theta,R\cup\{{\bm{c}}\},j} with

πj+​(k)=Mπj(k)+w𝒄𝕀[cj=k]M+w𝒄.\pi_{j}^{+}(k)=\frac{M\pi_{j}(k)+w_{{\bm{c}}}\mathbb{I}[c_{j}=k]}{M+w_{{\bm{c}}}}.

The candidate contributes mass to its assigned label at each position. The effect on uncertainty depends on where this support falls relative to the existing label masses. Two candidates with equal weights can therefore produce different changes in uncertainty. A weight-only top-kk rule ranks them independently of the current retained set. ACR evaluates their effects on the supervision already formed by that set, using the following change in the criterion.

For 𝒄∈CK∖R{\bm{c}}\in C_{K}\setminus R, define

Δ𝒄​Jθ​(R)\displaystyle\Delta_{{\bm{c}}}J_{\theta}(R) ≡Jθ​(R∪{𝒄},𝒙,y)−Jθ​(R,𝒙,y)\displaystyle\equiv J_{\theta}(R\cup\{{\bm{c}}\};{\bm{x}},y)-J_{\theta}(R;{\bm{x}},y)
=Uθ​(R∪{𝒄},𝒙,y)−Uθ​(R,𝒙,y)\displaystyle=\mathrm{U}_{\theta}(R\cup\{{\bm{c}}\};{\bm{x}},y)-\mathrm{U}_{\theta}(R;{\bm{x}},y)
−wθ​(𝒄∣𝒙,y).\displaystyle-w_{\theta}({\bm{c}}\mid{\bm{x}},y).

The first difference measures the change in retained ambiguity. The final term rewards the posterior mass recovered by 𝒄{\bm{c}}.

At each step, we select the candidate with the smallest Δ𝒄​Jθ​(R)\Delta_{{\bm{c}}}J_{\theta}(R). We add it if this value is negative and stop otherwise.

Algorithm 1 Greedy Candidate Retention
0:  Pool CKC_{K} and full-distribution weights wθ​(𝒄∣𝒙,y)w_{\theta}({\bm{c}}\mid{\bm{x}},y)
0:  Retained set RR
1:  Choose 𝒄0∈arg⁡max𝒄∈CK​wθ​(𝒄∣𝒙,y){\bm{c}}_{0}\in\arg\max_{{\bm{c}}\in C_{K}}w_{\theta}({\bm{c}}\mid{\bm{x}},y).
2:  Set R←{𝒄0}R\leftarrow\{{\bm{c}}_{0}\}.
3:  while R≠CKR\neq C_{K} do
4:   Choose
𝒄†∈arg⁡min𝒄∈CK∖R​Δ𝒄​Jθ​(R).{\bm{c}}^{\dagger}\in\arg\min_{{\bm{c}}\in C_{K}\setminus R}\Delta_{{\bm{c}}}J_{\theta}(R).
5:   if Δ𝒄†​Jθ​(R)≥0\Delta_{{\bm{c}}^{\dagger}}J_{\theta}(R)\geq 0 then
6:    break
7:   end if
8:   Update R←R∪{𝒄†}R\leftarrow R\cup\{{\bm{c}}^{\dagger}\}.
9:  end while
10:  return RR.

Each accepted step strictly lowers JθJ_{\theta} and adds one candidate, so the algorithm terminates at a one-addition local optimum over CKC_{K}. With maintained coordinate label masses, selection takes O⁡(N2​m)O(N^{2}m) time in the worst case, where N=|CK|≤KN=|C_{K}|\leq K. At round tt, the selected RtR_{t} defines the fixed targets in Section 3.2. Candidate weights and retention are recomputed after the neural update.

Table 1: Character accuracy on digit addition (%; mean ±\pm std). N/A: did not finish within 24 hours.
MNIST KMNIST
Method n=1n=1 n=2n=2 n=3n=3 n=4n=4 n=1n=1 n=2n=2 n=3n=3 n=4n=4
DeepStochLog 99.00±0.1399.00_{\pm\mathmakebox[l]{0.13}} 98.88±0.0298.88_{\pm\mathmakebox[l]{0.02}} 98.94±0.1598.94_{\pm\mathmakebox[l]{0.15}} 98.98±0.1398.98_{\pm\mathmakebox[l]{0.13}} 94.00±0.7394.00_{\pm\mathmakebox[l]{0.73}} 94.06±0.3094.06_{\pm\mathmakebox[l]{0.30}} 94.00±0.2994.00_{\pm\mathmakebox[l]{0.29}} 93.93±0.4993.93_{\pm\mathmakebox[l]{0.49}}
NeurASP 50.01±0.4450.01_{\pm\mathmakebox[l]{0.44}} 10.21±0.3910.21_{\pm\mathmakebox[l]{0.39}} 10.21±0.3910.21_{\pm\mathmakebox[l]{0.39}} N/A 10.01±0.5810.01_{\pm\mathmakebox[l]{0.58}} 10.05±0.5110.05_{\pm\mathmakebox[l]{0.51}} N/A N/A
DeepProbLog 97.42±0.2997.42_{\pm\mathmakebox[l]{0.29}} N/A N/A N/A 80.54±0.2180.54_{\pm\mathmakebox[l]{0.21}} N/A N/A N/A
ABL-hamming 98.49±0.2298.49_{\pm\mathmakebox[l]{0.22}} 98.65±0.0898.65_{\pm\mathmakebox[l]{0.08}} 98.63±0.1298.63_{\pm\mathmakebox[l]{0.12}} 60.41±40.2760.41_{\pm\mathmakebox[l]{40.27}} 91.48±0.5491.48_{\pm\mathmakebox[l]{0.54}} 92.43±0.6592.43_{\pm\mathmakebox[l]{0.65}} 63.51±36.9163.51_{\pm\mathmakebox[l]{36.91}} 47.74±37.1347.74_{\pm\mathmakebox[l]{37.13}}
ABL-confidence 79.31±41.3579.31_{\pm\mathmakebox[l]{41.35}} 79.32±41.6179.32_{\pm\mathmakebox[l]{41.61}} 77.10±43.5377.10_{\pm\mathmakebox[l]{43.53}} 79.22±41.6579.22_{\pm\mathmakebox[l]{41.65}} 86.32±25.4086.32_{\pm\mathmakebox[l]{25.40}} 79.80±34.9179.80_{\pm\mathmakebox[l]{34.91}} 72.76±38.7572.76_{\pm\mathmakebox[l]{38.75}} 54.84±42.9754.84_{\pm\mathmakebox[l]{42.97}}
A3​BL\text{A}^{3}\text{BL} 98.92±0.1598.92_{\pm\mathmakebox[l]{0.15}} 99.03±0.0999.03_{\pm\mathmakebox[l]{0.09}} 98.68±0.1798.68_{\pm\mathmakebox[l]{0.17}} 98.42±0.0898.42_{\pm\mathmakebox[l]{0.08}} 94.84±0.2594.84_{\pm\mathmakebox[l]{0.25}} 94.31±0.2194.31_{\pm\mathmakebox[l]{0.21}} 93.16±1.0093.16_{\pm\mathmakebox[l]{1.00}} 92.97±0.4492.97_{\pm\mathmakebox[l]{0.44}}
ACR (ours) 98.95±0.0498.95_{\pm\mathmakebox[l]{0.04}} 99.06±0.1899.06_{\pm\mathmakebox[l]{0.18}} 99.02±0.0799.02_{\pm\mathmakebox[l]{0.07}} 98.93±0.0898.93_{\pm\mathmakebox[l]{0.08}} 94.44±0.2894.44_{\pm\mathmakebox[l]{0.28}} 95.12±0.4195.12_{\pm\mathmakebox[l]{0.41}} 94.44±0.1894.44_{\pm\mathmakebox[l]{0.18}} 94.35±0.5094.35_{\pm\mathmakebox[l]{0.50}}
Figure 2: Training curves for four representative modulus pairs per dataset. Annotations show ACR’s accuracy gain in percentage points over the strongest baseline. Full results in Appendix C.

4 Experiments

We empirically evaluate ACR through two questions:

  • •

    How does candidate retention compare with the single-candidate and posterior-weighted policies when valid candidates disagree on hidden concepts?

  • •

    What roles do the two terms in the objective play, and how does the retained set change during training?

Section 4.1 compares retention policies on digit addition (Manhaeve et al.,, 2018) and aggregated mod addition (He and Li,, 2025). Section 4.2 examines the two objective terms through ablations and retention dynamics. Section 4.3 examines applications to chess attack (Dai et al.,, 2019) and judicial sentencing (Huang et al.,, 2020).

4.1 Learning from Ambiguous Explanations

We begin with two arithmetic settings where the same observed target admits multiple valid digit assignments.

In digit addition, each example contains two sequences of nn digit images whose hidden labels form two integers; training observes only their sum (Manhaeve et al.,, 2018). For n=1n=1, SUM([Uncaptioned image],[Uncaptioned image])=3 reveals the sum 3, while the image labels remain hidden. MNIST (LeCun et al.,, 1998) and KMNIST (Clanuwat et al.,, 2018) supply the images for n∈{1,2,3,4}n\in\{1,2,3,4\}. We report character accuracy on the recovered digit labels, averaged over five runs. As nn grows, more hidden decisions must agree with the same target.

For these arithmetic tasks, we compare ACR with the single-candidate ABL-hamming and ABL-confidence (Dai et al.,, 2019), and A3​BL\text{A}^{3}\text{BL}  (He et al., 2024b, ). Digit addition also includes DeepProbLog, NeurASP, and DeepStochLog (Manhaeve et al.,, 2018; Yang et al.,, 2020; Winters et al.,, 2022). For aggregated mod addition, the A3​BL\text{A}^{3}\text{BL} runs use a top-32 cap and temperature 0.20.2, retaining all candidates when at most 32 are available.

Both ACR and A3​BL\text{A}^{3}\text{BL} maintain high character accuracy across operand lengths, whereas the hard-selection methods have lower means or larger variation across runs (Table 1). On KMNIST, ACR exceeds A3​BL\text{A}^{3}\text{BL} for n≥2n\geq 2, with gains of 0.81–1.38 percentage points; A3​BL\text{A}^{3}\text{BL} performs better at n=1n=1.

Figure 3: Task-wise ablation distributions across 19 modulus pairs per dataset. Columns: character accuracy, TLM, and retained set size. ACR w/o mass and ACR w/o uncertainty (w/o M and w/o U in the panels) remove the mass and uncertainty terms.

Digit addition tests scaling in the number of hidden positions but uses a single rule per example. Aggregated mod addition introduces a harder coupling: a shared digit predictor must satisfy two modular rules simultaneously. For a modulus kk, each example contains two one-digit images with hidden labels z1,z2z_{1},z_{2} and one observed residue:

yk=(z1+z2)modk.y_{k}=(z_{1}+z_{2})\bmod k.

Each training example samples k∈{k1,k2}k\in\{k_{1},k_{2}\} and all examples share one digit predictor, so MODSUM_3([Uncaptioned image],[Uncaptioned image])=0 and MODSUM_4([Uncaptioned image],[Uncaptioned image])=3 follow different modular rules within the same training set. We evaluate 19 modulus pairs with 30,000 examples and three runs per pair (He and Li,, 2025).

The representative curves in Figure 2 reveal a difference in learning dynamics: ACR maintains high accuracy after the initial rise, while A3​BL\text{A}^{3}\text{BL} declines after an earlier peak in some settings. Across the evaluated modulus pairs, ACR improves concept accuracy over A3​BL\text{A}^{3}\text{BL} in most settings. The gains vary across pairs; Appendix C reports the complete results, including small or negative gains.

4.2 Understanding Candidate Retention

The retention objective JθJ_{\theta} balances retained uncertainty against discarded posterior mass. We study these roles through controlled ablations of the two terms and a training trace of the resulting supervision. Both use retained true-label weight (TLM) as a diagnostic. Let z⋆=(zj⋆)j=1mz^{\star}=(z_{j}^{\star})_{j=1}^{m} denote the ground-truth concepts; we define

TLMθ(R;𝒙,y)=1m∑j=1m∑𝒄∈Rw~θ(𝒄∣𝒙,y)𝕀[cj=zj⋆].\mathrm{TLM}_{\theta}(R;{\bm{x}},y)=\frac{1}{m}\sum_{j=1}^{m}\sum_{{\bm{c}}\in R}\widetilde{w}_{\theta}({\bm{c}}\mid{\bm{x}},y)\mathbb{I}[c_{j}=z_{j}^{\star}].

Here w~θ\widetilde{w}_{\theta} is the candidate distribution used in the experiments (Appendix B). TLM averages the retained weight assigned to the correct concept without renormalizing over RR; it therefore reflects both retained mass and label quality.

The ablations in Table 2 compare three retention policies under a shared training protocol. Without the mass term, the initial singleton already minimizes uncertainty, Uθ=0\mathrm{U}_{\theta}=0, so the strict-improvement rule accepts no additions. Without uncertainty, each positive-weight addition reduces discarded mass, and the rule retains the full search pool of min⁡(32,|𝒔|)\min(32,|{\bm{s}}|) candidates. Both controls use the same scoring and raw-weight supervision as ACR.

Averaged over the aggregated mod-addition settings, ACR reaches 94.98% concept accuracy, compared with 20.01% for ACR w/o mass and 65.22% for ACR w/o uncertainty (Table 2). Removing either term also lowers retained true-label weight. ACR retains fewer candidates on average than ACR w/o uncertainty while assigning more weight to correct labels. Figure 3 shows how these differences vary across modulus pairs.

Table 2: Objective ablation averaged over all aggregated mod-addition settings (19 modulus pairs ×\times 2 datasets). ACR achieves the highest accuracy and TLM; removing either term degrades both.
Method Acc. (%) TLM (%) Size
ACR w/o mass 20.01 7.88 1.00
ACR w/o uncertainty 65.22 40.15 18.76
ACR 94.98 94.25 16.68
Figure 4: Training dynamics on MNIST mod addition (3,4)(3,4). Top: retained uncertainty Uθ\mathrm{U}_{\theta} and discarded mass 1−Mθ1-M_{\theta}; bottom: TLM. Lines: medians; bands: interquartile ranges.

The representative trace in Figure 4 follows the retained supervision on MNIST mod addition (3,4)(3,4). Retained uncertainty rises early and then falls, while discarded mass approaches zero. In this trace, sharper coordinate targets coexist later in training with coverage of nearly all model-posterior mass.

4.3 Beyond Arithmetic Reasoning

We next test ACR on two non-arithmetic tasks: chess attack and judicial sentencing, comparing with ABL-hamming, ABL-confidence (Dai et al.,, 2019), and A3​BL\text{A}^{3}\text{BL}  (He et al., 2024b, ). In chess attack, MNIST images occupy board squares and map to six piece types (Dai et al.,, 2019). Training provides square locations and whether any pair attacks; piece labels remain hidden. We report attack and piece accuracy from a single run on about 60,000 training and 10,000 test boards. All methods share the split and 20-epoch schedule; ACR uses five warm-up epochs of A3​BL\text{A}^{3}\text{BL} supervision before switching to retention-based selection.

Judicial sentencing takes judgment facts as input and predicts discrete sentencing factors, which a knowledge base maps to the observed prison term (Huang et al.,, 2020). We follow the task setup of Huang et al. and report Macro-F1 and Micro-F1, averaged over five runs.

Table 3: Results beyond arithmetic. Best in bold.
Chess (%) Judicial
Method Attack Piece Macro-F1 Micro-F1
ABL-hamming 69.01 17.95 0.32470 0.70256
ABL-confidence 69.01 17.95 0.32470 0.70256
A3​BL\text{A}^{3}\text{BL} 97.53 87.19 0.83224 0.89574
ACR (ours) 99.31 98.29 0.83309 0.89870

Warm-started ACR improves more on piece identification than on attack prediction (Table 3). Both marginalization-based methods outperform hard selection, while the judicial task leaves less room for adaptive retention to differentiate from A3​BL\text{A}^{3}\text{BL} .

5 Related Work

In this section, we review the two lines of work most relevant to our method: neuro-symbolic learning and abductive learning.

5.1 Neuro-Symbolic Learning

Neuro-symbolic methods connect perception and reasoning through architectures, loss functions, or explicit inference. Early systems embedded declarative rules in connectionist architectures (Towell and Shavlik,, 1994; d’Avila Garcez et al.,, 2002). Logic-as-loss methods translate symbolic constraints into differentiable objectives (Xu et al.,, 2018; Roychowdhury et al.,, 2021); their behavior depends on the chosen relaxation (van Krieken et al.,, 2022; He et al., 2024a, ).

Hybrid systems retain an explicit symbolic reasoner. DeepProbLog extends ProbLog with neural predicates (Manhaeve et al.,, 2018; Manhaeve et al.,, 2021), NeurASP couples neural perception with answer-set programming (Yang et al.,, 2020), and DeepStochLog uses stochastic logic programs (Winters et al.,, 2022). They share a modular connection between neural perception and symbolic reasoning, with different mechanisms for exchanging information. In ABL, abduction returns explanations that become labels for the neural update. This feedback interface makes the candidate-to-supervision step explicit: several valid joint assignments can supply different labels for the same inputs. ACR studies which of these explanations should contribute to the returned supervision and how their labels combine into concept targets.

5.2 Abductive Learning

ABL turns abduction, inferring hypotheses that explain observations under background knowledge (Magnani,, 2009), into a learning mechanism whose central aim is to make perception and reasoning mutually beneficial (Zhou,, 2019; Dai et al.,, 2019). The neural model predicts concepts, abduction revises them under the knowledge base, and the revised candidates supervise the next neural update. When several candidates explain the same target, classical ABL selects the one closest to the current prediction, using Hamming distance or model confidence. This decision depends on model quality: at cold start, an unreliable prediction can select a poor pseudo-label, which subsequent updates may reinforce (Tao et al.,, 2024; He et al., 2024b, ). ABLSim mitigates this dependence with inter-sample similarity that guides consistency optimization before the classifier becomes reliable (Huang et al.,, 2021).

Subsequent work extends the loop along three axes. (i) Knowledge scope: GABL enables abduction from inexact ground propositions (Cai et al.,, 2021), and related methods learn from raw data, semi-supervised signals, or inaccurate rules (Dai and Muggleton,, 2021; Huang et al.,, 2020; Yang et al.,, 2024). The framework has also been applied to PDE discovery (Gao et al.,, 2025) and document recognition (Gao et al.,, 2024). (ii) Identifiability: formal analyses connect learning reliability to knowledge-base structure (Tao et al.,, 2024; He and Li,, 2025). (iii) Reasoning control: C-ABL partitions the knowledge base into sub-bases introduced progressively (Hu et al.,, 2025).

These advances improve what knowledge is available or how reasoning is structured, but leave a separate design choice: how much of the candidate posterior should return to the neural model as supervision. This choice matters because target-level consistency need not identify the intended latent concepts: different candidates can explain one target while assigning different concept labels, producing reasoning shortcuts (Marconato et al.,, 2023; Bortolotti et al.,, 2024). A3​BL\text{A}^{3}\text{BL} addresses this ambiguity by weighting every valid candidate under the current model and marginalizing their coordinate labels (He et al., 2024b, ), avoiding an early single-candidate commitment. Classical ABL and A3​BL\text{A}^{3}\text{BL} therefore define the single-candidate and marginalization endpoints. ACR develops the intermediate range through a greedy subset-selection criterion combining supervision sharpness and posterior-mass coverage.

6 Limitations and Future Directions

ACR operates on the candidate set returned by abduction. Enumeration can become the bottleneck when this space grows combinatorially. The criterion uses unit coefficients, and task-dependent weighting remains unexplored. Our comparisons use fixed KK and temperature; validation-selected settings and retained-size-matched controls would further characterize the benefit of the selection rule. The chess result also calls for repeated runs with matched warm-up schedules. Candidate retention alone does not establish identifiability of the intended concept semantics, which is central to reasoning shortcuts (Marconato et al.,, 2023).

Future directions include jointly generating and retaining explanations, learning the relative weighting of uncertainty and mass, and extending retention to inaccurate rules or partial labels.

7 Conclusion

Candidate retention shapes concept supervision through the labels that explanations support jointly. ACR evaluates each addition by the model mass it recovers and the change it makes to coordinate uncertainty. Our risk bound separates these two selection-dependent quantities from model mismatch; the greedy rule uses them to construct a retained set for each training pair. Experiments show concept-accuracy gains over single-candidate baselines and A3​BL\text{A}^{3}\text{BL} in most evaluated aggregated mod-addition settings. Objective ablations support the complementary roles of mass and uncertainty in retaining alternatives while keeping concept targets informative.

References

  • Bortolotti et al., (2024) Bortolotti, S., Marconato, E., Carraro, T., Morettin, P., van Krieken, E., Vergari, A., Teso, S., and Passerini, A. (2024). A neuro-symbolic benchmark suite for concept quality and reasoning shortcuts. In Advances in Neural Information Processing Systems, volume 37, pages 115861–115905, Vancouver, BC, Canada. Curran Associates, Inc. Datasets and Benchmarks Track.
  • Cai et al., (2021) Cai, L.-W., Dai, W.-Z., Huang, Y.-X., Li, Y.-F., Muggleton, S., and Jiang, Y. (2021). Abductive learning with ground knowledge base. In Proceedings of the 30th International Joint Conference on Artificial Intelligence, pages 1815–1821, Virtual Event / Montreal, Canada. International Joint Conferences on Artificial Intelligence.
  • Clanuwat et al., (2018) Clanuwat, T., Bober-Irizar, M., Kitamoto, A., Lamb, A., Yamamoto, K., and Ha, D. (2018). Deep learning for classical japanese literature. CoRR, abs/1812.01718.
  • Dai and Muggleton, (2021) Dai, W.-Z. and Muggleton, S. (2021). Abductive knowledge induction from raw data. In Proceedings of the 30th International Joint Conference on Artificial Intelligence, pages 1845–1851, Virtual Event / Montreal, Canada. International Joint Conferences on Artificial Intelligence.
  • Dai et al., (2019) Dai, W.-Z., Xu, Q., Yu, Y., and Zhou, Z.-H. (2019). Bridging machine learning and logical reasoning by abductive learning. In Advances in Neural Information Processing Systems, volume 32, pages 2815–2826, Vancouver, BC, Canada. Curran Associates, Inc.
  • d’Avila Garcez et al., (2002) d’Avila Garcez, A. S., Broda, K. B., and Gabbay, D. M. (2002). Neural-Symbolic Learning Systems: Foundations and Applications. Perspectives in Neural Computing. Springer-Verlag, London, UK.
  • Gao et al., (2025) Gao, E.-H., Ge, C., Jiang, Y., and Zhou, Z.-H. (2025). Discovering symbolic partial differential equation by abductive learning. In Advances in Neural Information Processing Systems, volume 38, pages 57359–57385, San Diego, CA, USA and Mexico City, Mexico. Curran Associates, Inc.
  • Gao et al., (2024) Gao, E.-H., Huang, Y.-X., Hu, W.-C., Zhu, X.-H., and Dai, W.-Z. (2024). Knowledge-enhanced historical document segmentation and recognition. In Proceedings of the AAAI Conference on Artificial Intelligence, volume 38, pages 8409–8416, Vancouver, BC, Canada. AAAI Press.
  • (9) He, H.-Y., Dai, W.-Z., and Li, M. (2024a). Reduced implication-bias logic loss for neuro-symbolic learning. Machine Learning, 113(6):3357–3377.
  • He and Li, (2025) He, H.-Y. and Li, M. (2025). A learnability analysis on neuro-symbolic learning. In Advances in Neural Information Processing Systems, volume 38, pages 84923–84956, San Diego, CA, USA and Mexico City, Mexico. Curran Associates, Inc.
  • (11) He, H.-Y., Sun, H., Xie, Z., and Li, M. (2024b). Ambiguity-aware abductive learning. In Proceedings of the 41st International Conference on Machine Learning, volume 235 of Proceedings of Machine Learning Research, pages 18019–18042, Vienna, Austria. PMLR.
  • Hu et al., (2025) Hu, W.-C., Li, Q.-J., Jia, L.-H., Ge, C., Li, Y.-F., Jiang, Y., and Zhou, Z.-H. (2025). Curriculum abductive learning. In Advances in Neural Information Processing Systems, volume 38, pages 174455–174483, San Diego, CA, USA and Mexico City, Mexico. Curran Associates, Inc.
  • Huang et al., (2021) Huang, Y.-X., Dai, W.-Z., Cai, L.-W., Muggleton, S. H., and Jiang, Y. (2021). Fast abductive learning by similarity-based consistency optimization. In Advances in Neural Information Processing Systems, volume 34, pages 26574–26584, Virtual Event. Curran Associates, Inc.
  • Huang et al., (2020) Huang, Y.-X., Dai, W.-Z., Yang, J., Cai, L.-W., Cheng, S., Huang, R., Li, Y.-F., and Zhou, Z.-H. (2020). Semi-supervised Abductive Learning and its application to theft judicial sentencing. In 2020 IEEE International Conference on Data Mining (ICDM), pages 1070–1075, Virtual Event / Sorrento, Italy. IEEE.
  • LeCun et al., (1998) LeCun, Y., Bottou, L., Bengio, Y., and Haffner, P. (1998). Gradient-based learning applied to document recognition. Proceedings of the IEEE, 86(11):2278–2324.
  • Lücke, (2019) Lücke, J. (2019). Truncated variational expectation maximization. arXiv:1610.03113, version 3.
  • Magnani, (2009) Magnani, L. (2009). Abductive Cognition: The Epistemological and Eco-Cognitive Dimensions of Hypothetical Reasoning, volume 3 of Cognitive Systems Monographs. Springer, Berlin, Heidelberg.
  • Manhaeve et al., (2018) Manhaeve, R., Dumančić, S., Kimmig, A., Demeester, T., and De Raedt, L. (2018). DeepProbLog: Neural probabilistic logic programming. In Advances in Neural Information Processing Systems, volume 31, pages 3753–3763, Montréal, QC, Canada. Curran Associates, Inc.
  • Manhaeve et al., (2021) Manhaeve, R., Dumančić, S., Kimmig, A., Demeester, T., and De Raedt, L. (2021). Neural probabilistic logic programming in DeepProbLog. Artificial Intelligence, 298:103504.
  • Marconato et al., (2023) Marconato, E., Teso, S., Vergari, A., and Passerini, A. (2023). Not all neuro-symbolic concepts are created equal: Analysis and mitigation of reasoning shortcuts. In Advances in Neural Information Processing Systems, volume 36, pages 72507–72539, New Orleans, LA, USA. Curran Associates, Inc.
  • Roychowdhury et al., (2021) Roychowdhury, S., Diligenti, M., and Gori, M. (2021). Regularizing deep networks with prior knowledge: A constraint-based approach. Knowledge-Based Systems, 222:106989.
  • Tao et al., (2024) Tao, L., Huang, Y.-X., Dai, W.-Z., and Jiang, Y. (2024). Deciphering raw data in neuro-symbolic learning with provable guarantees. In Proceedings of the AAAI Conference on Artificial Intelligence, volume 38, pages 15310–15318, Vancouver, BC, Canada. AAAI Press.
  • Towell and Shavlik, (1994) Towell, G. G. and Shavlik, J. W. (1994). Knowledge-based artificial neural networks. Artificial Intelligence, 70(1–2):119–165.
  • van Krieken et al., (2022) van Krieken, E., Acar, E., and van Harmelen, F. (2022). Analyzing differentiable fuzzy logic operators. Artificial Intelligence, 302:103602.
  • Winters et al., (2022) Winters, T., Marra, G., Manhaeve, R., and De Raedt, L. (2022). DeepStochLog: Neural stochastic logic programming. In Proceedings of the AAAI Conference on Artificial Intelligence, volume 36, pages 10090–10100, Virtual Event. AAAI Press.
  • Xu et al., (2018) Xu, J., Zhang, Z., Friedman, T., Liang, Y., and Van den Broeck, G. (2018). A semantic loss function for deep learning with symbolic knowledge. In Proceedings of the 35th International Conference on Machine Learning, volume 80 of Proceedings of Machine Learning Research, pages 5502–5511, Stockholm, Sweden. PMLR.
  • Yang et al., (2024) Yang, X.-W., Shao, J.-J., Tu, W.-W., Li, Y.-F., Dai, W.-Z., and Zhou, Z.-H. (2024). Safe abductive learning in the presence of inaccurate rules. In Proceedings of the AAAI Conference on Artificial Intelligence, volume 38, pages 16361–16369, Vancouver, BC, Canada. AAAI Press.
  • Yang et al., (2020) Yang, Z., Ishay, A., and Lee, J. (2020). NeurASP: Embracing neural networks into answer set programming. In Proceedings of the 29th International Joint Conference on Artificial Intelligence, pages 1755–1762, Virtual Event / Yokohama, Japan. International Joint Conferences on Artificial Intelligence.
  • Zhou, (2019) Zhou, Z.-H. (2019). Abductive Learning: Towards bridging machine learning and logical reasoning. Science China Information Sciences, 62(7):076101.

Appendix

Appendix A Proofs

A.1 Fixed-Support EM Connection

This subsection considers the product-posterior choice wθ​(𝒄∣𝒙,y)=pθ​(𝒄∣𝒙)/Zθ​(𝒙,y)w_{\theta}({\bm{c}}\mid{\bm{x}},y)=p_{\theta}({\bm{c}}\mid{\bm{x}})/Z_{\theta}({\bm{x}},y) and normalized retained targets. For a fixed positive-mass set RR, define

logZθ,R(𝒙,y)=log∑𝒄∈Rpθ(𝒄∣𝒙).\log Z_{\theta,R}({\bm{x}},y)=\log\sum_{{\bm{c}}\in R}p_{\theta}({\bm{c}}\mid{\bm{x}}).

The following identity is standard in truncated variational EM (Lücke,, 2019).

Proposition 1.

With product-posterior weights and RR fixed during an update, computing qt=qθt,Rq_{t}=q_{\theta_{t},R} performs the exact E-step for log⁡Zθ,R​(𝐱,y)\log Z_{\theta,R}({\bm{x}},y). Holding qtq_{t} fixed, an update that reduces

ℒret​(θ,qt)=−𝔼𝒄∼qt​[log⁡pθ​(𝒄∣𝒙)]{\mathcal{L}}_{\mathrm{ret}}(\theta;q_{t})=-{\mathbb{E}}_{{\bm{c}}\sim q_{t}}[\log p_{\theta}({\bm{c}}\mid{\bm{x}})]

performs a generalized M-step.

Proof of Proposition 1.

Proof.

For any distribution qq supported on RR, the standard variational identity gives

log⁡Zθ,R​(𝒙,y)=ℱ𝒙,y​(q,θ)+KL⁡(q∥qθ,R),\log Z_{\theta,R}({\bm{x}},y)=\mathcal{F}_{{\bm{x}},y}(q,\theta)+\mathrm{KL}\!\left(q\middle\|q_{\theta,R}\right),

where ℱ𝒙,y​(q,θ)=𝔼𝒄∼q​[log⁡pθ​(𝒄∣𝒙)]+H⁡(q)\mathcal{F}_{{\bm{x}},y}(q,\theta)={\mathbb{E}}_{{\bm{c}}\sim q}[\log p_{\theta}({\bm{c}}\mid{\bm{x}})]+H(q). At θ=θt\theta=\theta_{t}, setting q=qθt,Rq=q_{\theta_{t},R} makes the KL term zero and the bound tight: this is the exact E-step. With qtq_{t} fixed, H⁡(qt)H(q_{t}) is constant in θ\theta, so reducing ℒret​(θ,qt)=−𝔼qt​[log⁡pθ​(𝒄∣𝒙)]{\mathcal{L}}_{\mathrm{ret}}(\theta;q_{t})=-{\mathbb{E}}_{q_{t}}[\log p_{\theta}({\bm{c}}\mid{\bm{x}})] raises ℱ\mathcal{F} and thus the likelihood bound, giving a generalized M-step. ∎

The likelihood statement concerns a fixed support and these normalized product-posterior targets. The task-specific score and raw-weight convention used for aggregated mod addition are specified below.

Appendix B Experimental Protocol

Hardware.

All experiments were run on a server with two NVIDIA A100-PCIE GPUs (40 GB each) and two Intel Xeon Platinum 8358 CPUs.

Repeated runs.

Digit addition reports mean and standard deviation over five runs. Aggregated mod addition, its ablations, and retention dynamics each use three runs; Figure 4 shows medians and interquartile ranges. Chess attack uses one run with a 20-epoch schedule; ACR uses five warm-up epochs of A3​BL\text{A}^{3}\text{BL} supervision. Judicial sentencing averages over five runs.

Comparison methods.

Comparison methods use their official implementations, including the task-specific data splits and hyperparameters, following the evaluation protocols described above. For aggregated mod addition, ACR uses K=32K=32 and a temperature of 0.20.2, the same settings as A3​BL\text{A}^{3}\text{BL} . The chess warm-up schedule is stated alongside its results.

Reproducibility.

The supplementary materials include data-preprocessing, training, and analysis code, together with task-specific configurations and instructions for reproducing all reported results.

Appendix C Full Aggregated Mod-Addition Results

We follow the aggregated mod-addition setup of He and Li, (2025) and evaluate all 19 modulus pairs classified as learnable. Figures 5 and 6 report training trajectories for all pairs. ACR achieves higher accuracy than A3​BL\text{A}^{3}\text{BL} on most pairs in both datasets. The largest gap in favor of the baseline occurs at (2,7)(2,7), where A3​BL\text{A}^{3}\text{BL} leads by about three points.

Figure 5: Character-accuracy trajectories for all 19 MNIST modulus pairs. The bottom-right panel summarizes ACR’s test gain over the strongest baseline.
Figure 6: Character-accuracy trajectories for all 19 KMNIST modulus pairs. The bottom-right panel summarizes ACR’s test gain over the strongest baseline.