{"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-2602-04651","title":"SAFE: Stable Alignment Finetuning with Entropy-Aware Predictive Control for Reinforcement Learning from Human Feedback (RLHF)","arxiv_id":"2602.04651","date":"2026-02-04","proceeding":null,"authors":["Dipan Maity"],"abstract":"Proximal Policy Optimization (PPO) has been positioned by recent literature as the canonical method for the RL part of Reinforcement Learning from Human Feedback (RLHF). PPO performs well empirically but has a heuristic motivation and handles the KL-divergence constraint used in LM-RLHF in an ad-hoc manner and suffers form reward oscillations, entropy collapse, value function drift, and sudden policy divergence that require frequent restarts and extensive hyperparameter tuning. In this paper, we develop a new pure on policy actor-critic RL method for the LM-RLHF setting. We present SAFE (Stable Alignment Finetuning with Entropy-aware control),a novel RLHF algorithm that combines a Double Soft-Min Critic for pessimistic value estimation with a new multi-layer stabilization framework combining entropy-gated KL regulation, and PID-controlled adaptive thresholds. Unlike standard PPO's symmetric KL penalties, SAFE distinguishes high-entropy exploration from low-entropy mode collapse and adjusts penalties dynamically based on reward velocity. Experiments on a 3B parameter model show SAFE achieves +5.15\\% training-average reward than PPO (0.725 vs 0.689), negligible reward crashes, and superior KL control than ppo . Our method adds minimal computational overhead and provides an interpretable, crash-resistant RLHF framework that maintains aggressive learning speed while ensuring stable long-horizon optimization suitable for production deployment. Code is available at https://github.com/ryyzn9/SAFE","url_abs":"https://arxiv.org/abs/2602.04651","url_pdf":"https://arxiv.org/pdf/2602.04651","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=2602.04651","mcp":{"get_harvested_code_for_paper":{"arxiv_id":"2602.04651"}},"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/ryyzn9/SAFE","reach":null}],"summary":{"ran":7,"unverified":3},"by_repo_kind":{"found_in_text":{"samples":10,"ran":7,"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":10,"samples":[{"code_sha256_prefix":"348a8d18249af234","entry":"DoubleCritic","repo":"ryyzn9/SAFE","repo_kind":"found_in_text","path":"safe/trainers/safe.py","file_url":"https://github.com/ryyzn9/SAFE/blob/HEAD/safe/trainers/safe.py","link_basis":"first_harvest_node","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":"348a8d18249af234"}},{"code_sha256_prefix":"97aca6b496126665","entry":"EntropyAwarePredictiveController","repo":"ryyzn9/SAFE","repo_kind":"found_in_text","path":"safe/trainers/safe.py","file_url":"https://github.com/ryyzn9/SAFE/blob/HEAD/safe/trainers/safe.py","link_basis":"first_harvest_node","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":"97aca6b496126665"}},{"code_sha256_prefix":"48c8a6b260a13f7f","entry":"PIDController","repo":"ryyzn9/SAFE","repo_kind":"found_in_text","path":"safe/trainers/safe.py","file_url":"https://github.com/ryyzn9/SAFE/blob/HEAD/safe/trainers/safe.py","link_basis":"first_harvest_node","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":"48c8a6b260a13f7f"}},{"code_sha256_prefix":"3c780e38c4b7bf11","entry":"PhaseDetector","repo":"ryyzn9/SAFE","repo_kind":"found_in_text","path":"safe/trainers/safe.py","file_url":"https://github.com/ryyzn9/SAFE/blob/HEAD/safe/trainers/safe.py","link_basis":"first_harvest_node","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":"3c780e38c4b7bf11"}},{"code_sha256_prefix":"1944d530269a8eda","entry":"RewardNormalizer","repo":"ryyzn9/SAFE","repo_kind":"found_in_text","path":"safe/trainers/safe.py","file_url":"https://github.com/ryyzn9/SAFE/blob/HEAD/safe/trainers/safe.py","link_basis":"first_harvest_node","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":"1944d530269a8eda"}},{"code_sha256_prefix":"0f00da5a257590d7","entry":"S3KLQConfig","repo":"ryyzn9/SAFE","repo_kind":"found_in_text","path":"safe/trainers/safe.py","file_url":"https://github.com/ryyzn9/SAFE/blob/HEAD/safe/trainers/safe.py","link_basis":"first_harvest_node","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":"0f00da5a257590d7"}},{"code_sha256_prefix":"4067dfbd13ce4f31","entry":"SAFEConfig","repo":"ryyzn9/SAFE","repo_kind":"found_in_text","path":"safe/trainers/safe.py","file_url":"https://github.com/ryyzn9/SAFE/blob/HEAD/safe/trainers/safe.py","link_basis":"first_harvest_node","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":"4067dfbd13ce4f31"}},{"code_sha256_prefix":"2aab167c40842355","entry":"BaseKLController","repo":"ryyzn9/SAFE","repo_kind":"found_in_text","path":"safe/trainers/safe.py","file_url":"https://github.com/ryyzn9/SAFE/blob/HEAD/safe/trainers/safe.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":"2aab167c40842355"}},{"code_sha256_prefix":"5e6545e6f0ce1a83","entry":"BaseTrainer","repo":"ryyzn9/SAFE","repo_kind":"found_in_text","path":"safe/trainers/safe.py","file_url":"https://github.com/ryyzn9/SAFE/blob/HEAD/safe/trainers/safe.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":"5e6545e6f0ce1a83"}},{"code_sha256_prefix":"62b0dcfc9f2c25f0","entry":"SAFETrainer","repo":"ryyzn9/SAFE","repo_kind":"found_in_text","path":"safe/trainers/safe.py","file_url":"https://github.com/ryyzn9/SAFE/blob/HEAD/safe/trainers/safe.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":"62b0dcfc9f2c25f0"}}]},"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}