{"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-2607-06930","title":"Imputation Meets Clustering: Exploiting Latent Subgroup Structure for Missing Data Recovery","arxiv_id":"2607.06930","date":"2026-07-08","proceeding":null,"authors":["Chuyao Zhang","E Li","Taochen Chen","Yiqun Zhang","Yuzhu Ji","Shuping Zhao","Peng Liu","Yiu-ming Cheung"],"abstract":"Missing data is prevalent in practical applications, making effective imputation an essential preprocessing step for downstream analysis. Real-world datasets often exhibit complex latent structures composed of multiple subgroups with distinct distributions. However, existing methods often overlook such population heterogeneity. Without explicit structural guidance, these methods tend to produce generic estimates that blur subgroup boundaries and lack instance-level fidelity. While incorporating subgroup information offers a remedy, it faces a circular dependency: reliable subgroup identification requires complete data, while data completion is the imputation objective itself. To resolve this, we propose CAGI (Cluster-Aware Generative Imputation), a framework that reformulates clustering and imputation as a mutually reinforcing co-optimization process. CAGI employs a ``Partition-Guide-Restore'' strategy where dynamic cluster assignments act as local priors to condition a Generative Adversarial Network. An iterative feedback loop is established to progressively refine both cluster structures and imputed values toward faithful subgroup distributions. To ensure distributional stability, CAGI further employs a multi-level optimization objective combining instance-level reconstruction with distribution-level regularization. Extensive experiments on 14 benchmark datasets with 15 representative baselines demonstrate the superiority of CAGI. The source code is available at: https://github.com/supercocachii/CAGI","url_abs":"https://arxiv.org/abs/2607.06930","url_pdf":"https://arxiv.org/pdf/2607.06930","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=2607.06930","mcp":{"get_harvested_code_for_paper":{"arxiv_id":"2607.06930"}},"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/supercocachii/CAGI","reach":null}],"summary":{"ran":7,"unverified":1},"by_repo_kind":{"found_in_text":{"samples":8,"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":8,"samples":[{"code_sha256_prefix":"c085dd2e6df44854","entry":"analog_bits_to_int","repo":"supercocachii/CAGI","repo_kind":"found_in_text","path":"run_experiment.py","file_url":"https://github.com/supercocachii/CAGI/blob/HEAD/run_experiment.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NOASSERTION","inline_ok":false,"mcp_get_code":{"code_sha256":"c085dd2e6df44854"}},{"code_sha256_prefix":"7ecb6ba31d0c8202","entry":"binary_sampler","repo":"supercocachii/CAGI","repo_kind":"found_in_text","path":"utils.py","file_url":"https://github.com/supercocachii/CAGI/blob/HEAD/utils.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NOASSERTION","inline_ok":false,"mcp_get_code":{"code_sha256":"7ecb6ba31d0c8202"}},{"code_sha256_prefix":"366dc7c30002236f","entry":"cluster_onehot","repo":"supercocachii/CAGI","repo_kind":"found_in_text","path":"CAGI.py","file_url":"https://github.com/supercocachii/CAGI/blob/HEAD/CAGI.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NOASSERTION","inline_ok":false,"mcp_get_code":{"code_sha256":"366dc7c30002236f"}},{"code_sha256_prefix":"481292f0656764ac","entry":"get_num_bits","repo":"supercocachii/CAGI","repo_kind":"found_in_text","path":"run_experiment.py","file_url":"https://github.com/supercocachii/CAGI/blob/HEAD/run_experiment.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":true,"licence":"NOASSERTION","inline_ok":false,"mcp_get_code":{"code_sha256":"481292f0656764ac"}},{"code_sha256_prefix":"29f575f04acc7691","entry":"improved_noise_sampler","repo":"supercocachii/CAGI","repo_kind":"found_in_text","path":"CAGI.py","file_url":"https://github.com/supercocachii/CAGI/blob/HEAD/CAGI.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NOASSERTION","inline_ok":false,"mcp_get_code":{"code_sha256":"29f575f04acc7691"}},{"code_sha256_prefix":"a839ad767af99da1","entry":"int_to_analog_bits","repo":"supercocachii/CAGI","repo_kind":"found_in_text","path":"run_experiment.py","file_url":"https://github.com/supercocachii/CAGI/blob/HEAD/run_experiment.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NOASSERTION","inline_ok":false,"mcp_get_code":{"code_sha256":"a839ad767af99da1"}},{"code_sha256_prefix":"ade5323cb6088a6d","entry":"sample_batch_index","repo":"supercocachii/CAGI","repo_kind":"found_in_text","path":"utils.py","file_url":"https://github.com/supercocachii/CAGI/blob/HEAD/utils.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NOASSERTION","inline_ok":false,"mcp_get_code":{"code_sha256":"ade5323cb6088a6d"}},{"code_sha256_prefix":"3066702d867a6919","entry":"get_cluster_model","repo":"supercocachii/CAGI","repo_kind":"found_in_text","path":"CAGI.py","file_url":"https://github.com/supercocachii/CAGI/blob/HEAD/CAGI.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NOASSERTION","inline_ok":false,"mcp_get_code":{"code_sha256":"3066702d867a6919"}}]},"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}