{"about":{"site":"https://codewithpapers.app","non_affiliation":"Code with Papers and Syntology are not affiliated with, endorsed by, or sponsored by Papers with Code, Meta, or the pwc-archive mirror.","licence":"CC BY-SA 4.0","licence_url":"https://creativecommons.org/licenses/by-sa/4.0/legalcode","attribution":"https://codewithpapers.app/attribution","modified":"archive material modified by Syntology; see the attribution page"},"url":"/paper/arxiv-2601-21484","title":"ETS: Energy-Guided Test-Time Scaling for Training-Free RL Alignment","arxiv_id":"2601.21484","date":"2026-01-29","proceeding":"ICML","authors":["Xiuyu Li","Jinkai Zhang","Mingyang Yi","Yu Li","Longqiang Wang","Yue Wang","Ju Fan"],"abstract":"Reinforcement Learning (RL) post-training alignment for language models is effective, but also costly and unstable in practice, owing to its complicated training process. To address this, we propose a training-free inference method to sample directly from the optimal RL policy. The transition probability applied to Masked Language Modeling (MLM) consists of a reference policy model and an energy term. Based on this, our algorithm, Energy-Guided Test-Time Scaling (ETS), estimates the key energy term via online Monte Carlo, with a provable convergence rate. Moreover, to ensure practical efficiency, ETS leverages modern acceleration frameworks alongside tailored importance sampling estimators, substantially reducing inference latency while provably preserving sampling quality. Experiments on MLM (including autoregressive models and diffusion language models) across reasoning, coding, and science benchmarks show that our ETS consistently improves generation quality, validating its effectiveness and design. The code is available at https://github.com/sheriyuo/ETS.","url_abs":"https://arxiv.org/abs/2601.21484","url_pdf":"https://arxiv.org/pdf/2601.21484","source":{"archive":null,"snapshot":"2025-07-28","note":"not in the Papers with Code archive (frozen at the snapshot)","row_kind":"graph","title_abstract_authors_date":"arXiv metadata, CC0 1.0 (https://info.arxiv.org/help/license)"},"code_links":[],"tasks":[],"methods":[],"datasets_introduced":[],"methods_introduced":[],"results":[],"syntology":{"atlas_url":"https://app.syntology.ai/?focus=2601.21484","mcp":{"get_harvested_code_for_paper":{"arxiv_id":"2601.21484"}},"developers":"https://syntology.ai/developers","read_at":"2026-09-24T18:15:14+00:00","read_at_is":"when the build read Syntology's graph, not when any sample ran","claim":"Per-sample execution status on synthesized fixtures; not a correctness claim about the paper. Samples come from repositories linked to the paper, official or community; repo_kind says which.","repos":[{"mentioned_in_github":null,"is_official":null,"provenance":"deterministic:regex_extraction","mentioned_in_paper":null,"url":"https://github.com/sheriyuo/ETS","reach":null}],"summary":{"ran":1,"ran_draft_wrong":1,"unverified":14},"by_repo_kind":{"found_in_text":{"samples":16,"ran":2,"repositories":1}},"repo_kind_vocabulary":{"official":"The archive marks this repository official for the paper","named_in_paper":"The archive records that the paper mentions this repository; it is not marked official","listed":"In the archive's code links for this paper, not marked official and not recorded as mentioned in the paper","found_in_text":"Syntology found this repository in the paper's own text; whether it is the authors' implementation is not asserted","community":"Not in the archive's code links for this paper; a community repository Syntology harvested"},"n_pointer_only_for_licence":16,"samples":[{"code_sha256_prefix":"f9abf6d74699cc1c","entry":"causal_attention_bias","repo":"sheriyuo/ETS","repo_kind":"found_in_text","path":"llada/model/modeling_llada.py","file_url":"https://github.com/sheriyuo/ETS/blob/HEAD/llada/model/modeling_llada.py","link_basis":"harvester_set","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"f9abf6d74699cc1c"}},{"code_sha256_prefix":"52adf1b5d14ad2a3","entry":"remove_boxed","repo":"sheriyuo/ETS","repo_kind":"found_in_text","path":"aime24/utils.py","file_url":"https://github.com/sheriyuo/ETS/blob/HEAD/aime24/utils.py","link_basis":"harvester_set","language":"python","status":"ran_draft_wrong","verification_level":1,"contract_check":"OUTPUT_MISDECLARED","metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"52adf1b5d14ad2a3"}},{"code_sha256_prefix":"b95cadfa99b9514d","entry":"add_gumbel_noise","repo":"sheriyuo/ETS","repo_kind":"found_in_text","path":"llada/ets_is.py","file_url":"https://github.com/sheriyuo/ETS/blob/HEAD/llada/ets_is.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"b95cadfa99b9514d"}},{"code_sha256_prefix":"6b8a7622f02a3ba3","entry":"cal_log_probs","repo":"sheriyuo/ETS","repo_kind":"found_in_text","path":"llada/ets_is.py","file_url":"https://github.com/sheriyuo/ETS/blob/HEAD/llada/ets_is.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"6b8a7622f02a3ba3"}},{"code_sha256_prefix":"76a6a147b47bf42f","entry":"cal_log_probs","repo":"sheriyuo/ETS","repo_kind":"found_in_text","path":"llada/beam_search.py","file_url":"https://github.com/sheriyuo/ETS/blob/HEAD/llada/beam_search.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"76a6a147b47bf42f"}},{"code_sha256_prefix":"71625e4c2bfdcce7","entry":"compute_score","repo":"sheriyuo/ETS","repo_kind":"found_in_text","path":"llada/gsm8k.py","file_url":"https://github.com/sheriyuo/ETS/blob/HEAD/llada/gsm8k.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"71625e4c2bfdcce7"}},{"code_sha256_prefix":"3070c2dbc81462b4","entry":"entropy_reward","repo":"sheriyuo/ETS","repo_kind":"found_in_text","path":"llada/ets.py","file_url":"https://github.com/sheriyuo/ETS/blob/HEAD/llada/ets.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"3070c2dbc81462b4"}},{"code_sha256_prefix":"dd675dbd0c1a342e","entry":"extract_solution","repo":"sheriyuo/ETS","repo_kind":"found_in_text","path":"llada/gsm8k.py","file_url":"https://github.com/sheriyuo/ETS/blob/HEAD/llada/gsm8k.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"dd675dbd0c1a342e"}},{"code_sha256_prefix":"5bf328ff54a06f66","entry":"generate_candidate_blocks","repo":"sheriyuo/ETS","repo_kind":"found_in_text","path":"qwen/generate.py","file_url":"https://github.com/sheriyuo/ETS/blob/HEAD/qwen/generate.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"5bf328ff54a06f66"}},{"code_sha256_prefix":"053040ca62d36670","entry":"gsm8k_extract","repo":"sheriyuo/ETS","repo_kind":"found_in_text","path":"llada/extract_fn.py","file_url":"https://github.com/sheriyuo/ETS/blob/HEAD/llada/extract_fn.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"053040ca62d36670"}},{"code_sha256_prefix":"2ef5d6b4a02fe907","entry":"gumbel_max_sample","repo":"sheriyuo/ETS","repo_kind":"found_in_text","path":"llada/ets_is.py","file_url":"https://github.com/sheriyuo/ETS/blob/HEAD/llada/ets_is.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"2ef5d6b4a02fe907"}},{"code_sha256_prefix":"f1e5e614bcc0a4f7","entry":"humaneval_extract","repo":"sheriyuo/ETS","repo_kind":"found_in_text","path":"llada/extract_fn.py","file_url":"https://github.com/sheriyuo/ETS/blob/HEAD/llada/extract_fn.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"f1e5e614bcc0a4f7"}},{"code_sha256_prefix":"6d0bdd1050a5f9b9","entry":"last_boxed_only_string","repo":"sheriyuo/ETS","repo_kind":"found_in_text","path":"aime24/utils.py","file_url":"https://github.com/sheriyuo/ETS/blob/HEAD/aime24/utils.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"6d0bdd1050a5f9b9"}},{"code_sha256_prefix":"f8eed3cb436fe2a3","entry":"normalize_final_answer","repo":"sheriyuo/ETS","repo_kind":"found_in_text","path":"aime24/utils.py","file_url":"https://github.com/sheriyuo/ETS/blob/HEAD/aime24/utils.py","link_basis":"harvester_set","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"f8eed3cb436fe2a3"}},{"code_sha256_prefix":"a1e20a66389a45ff","entry":"scaled_dot_product_attention","repo":"sheriyuo/ETS","repo_kind":"found_in_text","path":"llada/model/modeling_llada.py","file_url":"https://github.com/sheriyuo/ETS/blob/HEAD/llada/model/modeling_llada.py","link_basis":"harvester_set","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"a1e20a66389a45ff"}},{"code_sha256_prefix":"fa5f0460d65ac3e8","entry":"self_certainty_reward","repo":"sheriyuo/ETS","repo_kind":"found_in_text","path":"llada/ets.py","file_url":"https://github.com/sheriyuo/ETS/blob/HEAD/llada/ets.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"fa5f0460d65ac3e8"}}]},"arxiv_metadata":{"licence":"arXiv metadata, CC0 1.0 (https://info.arxiv.org/help/license)","fields":["title","abstract","authors","date"],"primary_category":"cs.LG","source":"arxiv_2026.jsonl"},"syntology_extracted_results":null}