Procedure planning seeks to estimate a sequence of actions to transition from an observed initial state to a given goal state. Current procedure planning approaches directly predict action sequences from latent representations using feed-forward neural networks or diffusion-based inference. These paradigms treat every action as plausible, lacking the ability to enforce task-specific logical constraints that render certain actions irrelevant or not plausible. We propose CEFITO, a procedure planning approach that learns a predictor to express an action-conditioned representation space. Based on this representation space, we formulate procedure planning as a task-constrained optimization problem. Unlike prior methods, CEFITO explicitly reasons over the action space by omitting irrelevant actions during inference-time planning. This reformulation enables effective procedure planning and achieves state-of-the-art accuracy on two established procedure planning benchmarks.
Figures & tables
Figure 1: Overview of CEFITO. Left: Given an initial observation and a sequence of actions, our method maps action sequences to a predicted goal latent. The distance to the encoded goal latent defines the candidate’s energy, which we minimize at inference. Right: We learn a task-conditioned energy field via contrastive learning. At inference, CEFITO predicts the task ( e.g. , “Grill steak”), restricting the search to an action subset (pink region). The action sequence with minimal energy is the predicted plan.
Figure 2: CEFITO training framework. During training, a positive and N negative (wrong) action sequences are provided. Each sequence is tokenized using a text encoder and fed into the predictor Pθ , alongside the initial latent state xs . For each sequence we predict a goal state x~g , while predictions of negative sequences x~g,i− are pushed away from the target x~g and for positive sequences x~g+ pulled together. An auxiliary loss Laux supervises auxiliary latents of Pθ via a decoding head Dϕ .
Figure 3: CEFITO inference. A task classifier predicts c^ from (xs,xg) and restricts the search to A(c^) . The predictor Pθ scores every candidate sequence in A(c^) . The sequence approximating xg best, forms the action sequence prediction a~1:T .
Method
T=3
T=4
SR ↑
mAcc ↑
mIoU ↑
SR ↑
mAcc ↑
mIoU ↑
WLTDO [ 17 ]
1.87
21.64
31.70
0.77
17.92
26.43
UAAA [ 18 ]
2.15
20.21
30.87
0.98
19.86
27.09
UPN [ 50 ]
2.89
24.39
31.56
1.19
21.59
27.85
DDN [ 12 ]
12.18
31.29
47.48
5.97
27.10
48.46
PlaTe [ 51 ]
16.00
36.17
65.91
14.00
35.29
55.36
Table 1: Results on CrossTask [ 65 ] . We compare CEFITO with the state of the art on CrossTask val., using different planning horizons, and report SR, mACC, and mIoU (all in %, ↑ ). Best results highlighted in red ■■ ; second-best results in orange ■■ . For completeness, we report baselines not adhering to the protocol by [ 49 ] in gray ■■ .
Method
T=3
T=4
SR ↑
mAcc ↑
mIoU ↑
SR ↑
mAcc ↑
mIoU ↑
DDN [ 12 ]
13.90
20.19
64.78
11.13
17.71
68.06
P 3 IV [ 62 ]
15.40
21.67
76.31
11.32
18.85
70.53
EGPP [ 54 ]
19.57
31.42
84.95
13.59
26.72
84.72
Qwen2.5-VL-32B [ 3 ]
3.65
17.51
52.10
1.87
17.05
55.66
Qwen2.5-32B [ 60 ]
14.97
36.34
78.74
4.98
27.45
71.64
Table 2: Results on COIN [ 52 ] . We compare CEFITO with the state of the art on COIN val., using different planning horizons, and report SR, mACC, and mIoU (all in %, ↑ ). Best results are highlighted in red ■■ and second-best results in orange ■■ . For completeness, we report baselines not adhering to the protocol by [ 49 ] in gray ■■ .
Configuration
T=3
T=4
SR ↑
mAcc ↑
mIoU ↑
SR ↑
mAcc ↑
mIoU ↑
ViterbiPlanNet [ 49 ]
38.45
63.07
83.89
24.64
57.00
81.18
Baseline
13.65
47.82
69.53
9.24
35.44
67.76
+ triplet contrastive loss
37.58
63.21
83.67
23.54
56.18
80.91
+ adaptive margin
38.74
63.79
84.02
24.23
57.01
81.22
+ auxiliary regualrization (CEFITO)
39.62
64.12
84.29
24.76
57.53
81.58
Table 3: CEFITO training analysis. Component-wise analysis of CEFITO on CrossTask val. and report SR, mACC & mIoU (all in %, ↑ ). We add components to the previous row until achieving CEFITO. For reference, we include the recent state-of-the-art ViterbiPlanNet. Best results in red ■■ ; second-best results in orange ■■ .
Text Encoder
T=3
T=4
SR ↑
mAcc ↑
mIoU ↑
SR ↑
mAcc ↑
mIoU ↑
Random embedding
35.10
61.20
82.50
21.30
54.10
79.80
Flan-T5-base [ 14 ]
37.80
63.00
83.60
23.10
56.20
80.90
CLIP [ 46 ]
39.62
64.12
84.29
24.76
57.53
81.58
Table 4: Text encoder analysis. We analyze the choice of text encoder for CEFITO on CrossTask val. and report SR, mACC, and mIoU (all in %, ↑ ). Best results are highlighted in red ■■ and second-best results in orange ■■ .
Figure 4: Negative sampling analysis . We analyze the impact of the number of negative samples N (left) and the hard/easy negative sample ratio r (right) on CrossTask with \rm T=\text{3} using SR (in %, ↑ ). For r=\text{0} , only easy negatives are used. Vice versa , r=\text{1} generates only hard negatives. We indicate our default values in red ■■ .
Table 9
Figure 5: Inference runtime results . Left: We report inference runtime (in ms, ↓ ) over different planning horizons for inferring a single action sequence on the CrossTask dataset [ 65 ] . All runtimes are reported using the same hardware ( single A100 80GB GPU). Right: We report the average number of forward passes through our predictor Pθ required for inference. While CEFITO (in ■■ ) provides a comparable inference runtime for T≤4 to diffusion (PDPP [ 55 ] in ■■ ) and feed-forward (ViterbiPlanNet [ 49 ] in ■■ ) approaches, the runtime is significantly worse for T≥5 . For reference, we also report CEFITO without constraining the search space, indicated using ⋆ .
Method
T=3
T=4
SR ↑
mAcc ↑
mIoU ↑
SR ↑
mAcc ↑
mIoU ↑
Qwen2.5-VL-32B [ 3 ]
7.41
27.65
59.73
5.26
28.84
60.21
Qwen2.5-32B [ 60 ]
24.07
43.46
71.88
23.25
41.89
73.91
Gemini 2.5 Pro [ 21 ]
24.07
43.46
71.86
22.37
40.35
73.05
Qwen3-30B [ 59 ]
24.81
42.84
70.80
22.37
41.23
73.90
Qwen3-30B + PKG [ 49 ]
25.19
43.95
71.98
21.93
41.67
74.43
Table 7: Results on NIV [ 1 ] . We compare CEFITO with the state-of-the-art on NIV val., using different planning horizons, and report SR, mACC, and mIoU (all in %, ↑ ). Best results are highlighted in red ■■ and second-best results in orange ■■ .
Appendix figures & tables5 assets
Supplementary material from the paper’s appendix.
Appendix
Figure A.1: Auxiliary action sequence reconstruction. During training, the predictor Pθ is applied twice with shared weights: first , given xs , action tokens a1:4 , and masked state tokens x1:4mask , Pθ predicts the intermediate and goal latents x~1:3,x~g , respectivally. Second , given xs , xg , Pθ predicts the masked action tokens. These are supervised by the ground-truth action sequence a1:4 using the auxiliary loss Laux .
Figure A.2: Qualitative examples of CEFITO on CrossTask. Procedural video results where the top-ranked candidate plan (highlighted in ■■ ) matches the ground-truth action sequence, retrieved from the top-5 candidates ranked by L2 distance to the goal embedding xg .
Figure A.3: Qualitative examples of CEFITO on CrossTask. Extension of Fig. A.2 .
Figure A.4: Failure cases of CEFITO on CrossTask. Procedural video results where the top-ranked candidate plan (highlighted in ■■ ) fails to match the ground-truth action sequence (highlighted in ■■ ). For some examples, the ground truth falls outside of the top- 5 candidates.
Figure A.5: Failure cases of CEFITO on CrossTask. Extension of Fig. A.4 .