{"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-00534","title":"AIRE-Prune: Asymptotic Impulse-Response Energy for State Pruning in State Space Models","arxiv_id":"2602.00534","date":"2026-01-31","proceeding":null,"authors":["Apurba Prasad Padhy","Fernando Camacho","Saibal Mukhopadhyay"],"abstract":"State space models (SSMs) often sacrifice capacity, search space, or stability to offset the memory and compute costs of large state dimensions. We introduce a structured post-training pruning method for SSMs -- AIRE-Prune (Asymptotic Impulse-Response Energy for State PRUN(E)) -- that reduces each layer's state dimension by directly minimizing long-run output-energy distortion. AIRE-Prune assigns every state a closed-form asymptotic impulse-response energy-based score, i.e., the total impulse-response energy it contributes over an infinite horizon (time), and normalizes these scores layer-wise to enable global cross-layer comparison and selection. This extends modal truncation from single systems to deep stacks and aligns pruning with asymptotic response energy rather than worst-case gain. Across diverse sequence benchmarks, AIRE-Prune reveals substantial redundancy in SISO and MIMO SSMs with average pruning of 60.8%, with average accuracy drop of 0.29% without retraining, while significantly lowering compute. Code: https://github.com/falcon-arrow/AIRE-Prune.","url_abs":"https://arxiv.org/abs/2602.00534","url_pdf":"https://arxiv.org/pdf/2602.00534","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.00534","mcp":{"get_harvested_code_for_paper":{"arxiv_id":"2602.00534"}},"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/falcon-arrow/AIRE-Prune","reach":null}],"summary":{"ran_fixture":1,"ran":3,"unverified":14},"by_repo_kind":{"found_in_text":{"samples":18,"ran":4,"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":0,"samples":[{"code_sha256_prefix":"b4fc72406d2ff673","entry":"binary_operator","repo":"falcon-arrow/AIRE-Prune","repo_kind":"found_in_text","path":"s5/ssm.py","file_url":"https://github.com/falcon-arrow/AIRE-Prune/blob/HEAD/s5/ssm.py","link_basis":"plan_row","language":"python","status":"ran_fixture","verification_level":1,"contract_check":"RAISES","metamorphic_tier":"well_formed","behaviour_fingerprint":false,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"b4fc72406d2ff673"}},{"code_sha256_prefix":"ce310b19a858070f","entry":"dequantize","repo":"falcon-arrow/AIRE-Prune","repo_kind":"found_in_text","path":"s5/dataloaders/audio.py","file_url":"https://github.com/falcon-arrow/AIRE-Prune/blob/HEAD/s5/dataloaders/audio.py","link_basis":"harvester_set","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":true,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"ce310b19a858070f"}},{"code_sha256_prefix":"1b76e4737067023f","entry":"minmax_scale","repo":"falcon-arrow/AIRE-Prune","repo_kind":"found_in_text","path":"s5/dataloaders/audio.py","file_url":"https://github.com/falcon-arrow/AIRE-Prune/blob/HEAD/s5/dataloaders/audio.py","link_basis":"harvester_set","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":true,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"1b76e4737067023f"}},{"code_sha256_prefix":"1631cd936e9b1a90","entry":"quantize","repo":"falcon-arrow/AIRE-Prune","repo_kind":"found_in_text","path":"s5/dataloaders/audio.py","file_url":"https://github.com/falcon-arrow/AIRE-Prune/blob/HEAD/s5/dataloaders/audio.py","link_basis":"harvester_set","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"1631cd936e9b1a90"}},{"code_sha256_prefix":"faeebb869aee2124","entry":"cosine_annealing","repo":"falcon-arrow/AIRE-Prune","repo_kind":"found_in_text","path":"s5/train_helpers.py","file_url":"https://github.com/falcon-arrow/AIRE-Prune/blob/HEAD/s5/train_helpers.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"faeebb869aee2124"}},{"code_sha256_prefix":"476b5f7f116d7c9b","entry":"create_lra_imdb_classification_dataset","repo":"falcon-arrow/AIRE-Prune","repo_kind":"found_in_text","path":"s5/dataloading.py","file_url":"https://github.com/falcon-arrow/AIRE-Prune/blob/HEAD/s5/dataloading.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"476b5f7f116d7c9b"}},{"code_sha256_prefix":"1ee8fd3237e3786e","entry":"create_lra_listops_classification_dataset","repo":"falcon-arrow/AIRE-Prune","repo_kind":"found_in_text","path":"s5/dataloading.py","file_url":"https://github.com/falcon-arrow/AIRE-Prune/blob/HEAD/s5/dataloading.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"1ee8fd3237e3786e"}},{"code_sha256_prefix":"25edc240223b76c0","entry":"deprecated","repo":"falcon-arrow/AIRE-Prune","repo_kind":"found_in_text","path":"s5/dataloaders/base.py","file_url":"https://github.com/falcon-arrow/AIRE-Prune/blob/HEAD/s5/dataloaders/base.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"25edc240223b76c0"}},{"code_sha256_prefix":"c0135e087b049ad1","entry":"discretize_bilinear","repo":"falcon-arrow/AIRE-Prune","repo_kind":"found_in_text","path":"s5/ssm.py","file_url":"https://github.com/falcon-arrow/AIRE-Prune/blob/HEAD/s5/ssm.py","link_basis":"plan_row","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"c0135e087b049ad1"}},{"code_sha256_prefix":"f9485e8e0d1e9aee","entry":"discretize_zoh","repo":"falcon-arrow/AIRE-Prune","repo_kind":"found_in_text","path":"s5/ssm.py","file_url":"https://github.com/falcon-arrow/AIRE-Prune/blob/HEAD/s5/ssm.py","link_basis":"plan_row","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"f9485e8e0d1e9aee"}},{"code_sha256_prefix":"a2cd676d4d4c4910","entry":"linear_warmup","repo":"falcon-arrow/AIRE-Prune","repo_kind":"found_in_text","path":"s5/train_helpers.py","file_url":"https://github.com/falcon-arrow/AIRE-Prune/blob/HEAD/s5/train_helpers.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"a2cd676d4d4c4910"}},{"code_sha256_prefix":"72d464924b5dee96","entry":"listops_tokenizer","repo":"falcon-arrow/AIRE-Prune","repo_kind":"found_in_text","path":"s5/dataloaders/lra.py","file_url":"https://github.com/falcon-arrow/AIRE-Prune/blob/HEAD/s5/dataloaders/lra.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"72d464924b5dee96"}},{"code_sha256_prefix":"2b5694fa1220f8bd","entry":"make_DPLR_HiPPO","repo":"falcon-arrow/AIRE-Prune","repo_kind":"found_in_text","path":"s5/ssm_init.py","file_url":"https://github.com/falcon-arrow/AIRE-Prune/blob/HEAD/s5/ssm_init.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"2b5694fa1220f8bd"}},{"code_sha256_prefix":"3f26e3f7ea0487f3","entry":"make_HiPPO","repo":"falcon-arrow/AIRE-Prune","repo_kind":"found_in_text","path":"s5/ssm_init.py","file_url":"https://github.com/falcon-arrow/AIRE-Prune/blob/HEAD/s5/ssm_init.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"3f26e3f7ea0487f3"}},{"code_sha256_prefix":"0b867bda221aedfc","entry":"make_NPLR_HiPPO","repo":"falcon-arrow/AIRE-Prune","repo_kind":"found_in_text","path":"s5/ssm_init.py","file_url":"https://github.com/falcon-arrow/AIRE-Prune/blob/HEAD/s5/ssm_init.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"0b867bda221aedfc"}},{"code_sha256_prefix":"508a42e36707cd15","entry":"make_data_loader","repo":"falcon-arrow/AIRE-Prune","repo_kind":"found_in_text","path":"s5/dataloading.py","file_url":"https://github.com/falcon-arrow/AIRE-Prune/blob/HEAD/s5/dataloading.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"508a42e36707cd15"}},{"code_sha256_prefix":"261d3ddc455af2a0","entry":"masked_meanpool","repo":"falcon-arrow/AIRE-Prune","repo_kind":"found_in_text","path":"s5/seq_model.py","file_url":"https://github.com/falcon-arrow/AIRE-Prune/blob/HEAD/s5/seq_model.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"261d3ddc455af2a0"}},{"code_sha256_prefix":"2802de0d2519b69a","entry":"reduce_lr_on_plateau","repo":"falcon-arrow/AIRE-Prune","repo_kind":"found_in_text","path":"s5/train_helpers.py","file_url":"https://github.com/falcon-arrow/AIRE-Prune/blob/HEAD/s5/train_helpers.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"2802de0d2519b69a"}}]},"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}