{"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-00333","title":"Efficient and accurate steering of Large Language Models through attention-guided feature learning","arxiv_id":"2602.00333","date":"2026-01-30","proceeding":null,"authors":["Parmida Davarmanesh","Ashia Wilson","Adityanarayanan Radhakrishnan"],"abstract":"Steering, or direct manipulation of internal activations to guide LLM responses toward specific semantic concepts, is emerging as a promising avenue for both understanding how semantic concepts are stored within LLMs and advancing LLM capabilities. Yet, existing steering methods are remarkably brittle, with seemingly non-steerable concepts becoming completely steerable based on subtle algorithmic choices in how concept-related features are extracted. In this work, we introduce an attention-guided steering framework that overcomes three core challenges associated with steering: (1) automatic selection of relevant token embeddings for extracting concept-related features; (2) accounting for heterogeneity of concept-related features across LLM activations; and (3) identification of layers most relevant for steering. Across a steering benchmark of 512 semantic concepts, our framework substantially improved steering over previous state-of-the-art (nearly doubling the number of successfully steered concepts) across model architectures and sizes (up to 70 billion parameter models). Furthermore, we use our framework to shed light on the distribution of concept-specific features across LLM layers. Overall, our framework opens further avenues for developing efficient, highly-scalable fine-tuning algorithms for industry-scale LLMs.","url_abs":"https://arxiv.org/abs/2602.00333","url_pdf":"https://arxiv.org/pdf/2602.00333","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.00333","mcp":{"get_harvested_code_for_paper":{"arxiv_id":"2602.00333"}},"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/pdavar/attention_guided_steering","reach":null}],"summary":{"unverified":14},"by_repo_kind":{"found_in_text":{"samples":14,"ran":0,"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":14,"samples":[{"code_sha256_prefix":"262ae9648ea39b85","entry":"euclidean_distances_M_2","repo":"pdavar/attention_guided_steering","repo_kind":"found_in_text","path":"rfm.py","file_url":"https://github.com/pdavar/attention_guided_steering/blob/HEAD/rfm.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":"262ae9648ea39b85"}},{"code_sha256_prefix":"4180eb70e66a9fd2","entry":"fears_dataset_paired","repo":"pdavar/attention_guided_steering","repo_kind":"found_in_text","path":"datasets.py","file_url":"https://github.com/pdavar/attention_guided_steering/blob/HEAD/datasets.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":"4180eb70e66a9fd2"}},{"code_sha256_prefix":"b9e45af5e992c282","entry":"generate_on_text","repo":"pdavar/attention_guided_steering","repo_kind":"found_in_text","path":"generation_utils.py","file_url":"https://github.com/pdavar/attention_guided_steering/blob/HEAD/generation_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":"b9e45af5e992c282"}},{"code_sha256_prefix":"1471c86bd9c44663","entry":"generate_on_text_with_attn","repo":"pdavar/attention_guided_steering","repo_kind":"found_in_text","path":"generation_utils.py","file_url":"https://github.com/pdavar/attention_guided_steering/blob/HEAD/generation_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":"1471c86bd9c44663"}},{"code_sha256_prefix":"381afa5a5a166ee1","entry":"get_hidden_states_and_attns","repo":"pdavar/attention_guided_steering","repo_kind":"found_in_text","path":"direction_utils.py","file_url":"https://github.com/pdavar/attention_guided_steering/blob/HEAD/direction_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":"381afa5a5a166ee1"}},{"code_sha256_prefix":"d83ae6402143e959","entry":"get_prefix_attn_sum_for_layer_lastN","repo":"pdavar/attention_guided_steering","repo_kind":"found_in_text","path":"direction_utils.py","file_url":"https://github.com/pdavar/attention_guided_steering/blob/HEAD/direction_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":"d83ae6402143e959"}},{"code_sha256_prefix":"0fe18c53664a2dfd","entry":"get_prefix_attn_sum_for_layer_singletoken","repo":"pdavar/attention_guided_steering","repo_kind":"found_in_text","path":"direction_utils.py","file_url":"https://github.com/pdavar/attention_guided_steering/blob/HEAD/direction_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":"0fe18c53664a2dfd"}},{"code_sha256_prefix":"c94e1640cdcc4621","entry":"get_userprompt_start","repo":"pdavar/attention_guided_steering","repo_kind":"found_in_text","path":"generation_utils.py","file_url":"https://github.com/pdavar/attention_guided_steering/blob/HEAD/generation_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":"c94e1640cdcc4621"}},{"code_sha256_prefix":"c37898947e173254","entry":"jailbreaking_dataset","repo":"pdavar/attention_guided_steering","repo_kind":"found_in_text","path":"datasets.py","file_url":"https://github.com/pdavar/attention_guided_steering/blob/HEAD/datasets.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":"c37898947e173254"}},{"code_sha256_prefix":"9baa7adbc7a595e9","entry":"jailbreaking_dataset_paired","repo":"pdavar/attention_guided_steering","repo_kind":"found_in_text","path":"datasets.py","file_url":"https://github.com/pdavar/attention_guided_steering/blob/HEAD/datasets.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":"9baa7adbc7a595e9"}},{"code_sha256_prefix":"88ba8186492f0a3a","entry":"laplacian_M_2","repo":"pdavar/attention_guided_steering","repo_kind":"found_in_text","path":"rfm.py","file_url":"https://github.com/pdavar/attention_guided_steering/blob/HEAD/rfm.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":"88ba8186492f0a3a"}},{"code_sha256_prefix":"50d7f5a95721461b","entry":"laplacian_M_3","repo":"pdavar/attention_guided_steering","repo_kind":"found_in_text","path":"rfm.py","file_url":"https://github.com/pdavar/attention_guided_steering/blob/HEAD/rfm.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":"50d7f5a95721461b"}},{"code_sha256_prefix":"77f7e913fab304f1","entry":"minmax_normalize","repo":"pdavar/attention_guided_steering","repo_kind":"found_in_text","path":"control_toolkits.py","file_url":"https://github.com/pdavar/attention_guided_steering/blob/HEAD/control_toolkits.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":"77f7e913fab304f1"}},{"code_sha256_prefix":"a42bbfb14a05ac2f","entry":"split_indices","repo":"pdavar/attention_guided_steering","repo_kind":"found_in_text","path":"control_toolkits.py","file_url":"https://github.com/pdavar/attention_guided_steering/blob/HEAD/control_toolkits.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":"a42bbfb14a05ac2f"}}]},"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":{"kind":"leaderboard_placements","source":"Syntology's leaderboard-shaped extractor over the paper's own arXiv-HTML tables: a model pointed at a cell, the number was read from that cell and checked against the board's metric, dataset, split and scale, and an independent check accepted the entry; not reviewed by the paper's authors or the archive's editors","extractor_model":"global.anthropic.claude-sonnet-4-5-20250929-v1:0","verifier_model":null,"prompt_sha":"fa63d4bb9d755694","coverage":{"sentence":"Syntology has checked 6,264 of the 9,581 papers on this site that are newer than the archive; results from the others appear after they are checked.","papers_newer_than_archive":9581,"papers_checked":6264},"entries":[],"not_placed":{"boards":0,"rejected_by_independent_check":0,"refused_by_a_rule":0,"check_did_not_answer":0,"proposed_without_a_cell":0,"declined_by_site":0}}}