{"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/variational-imbalanced-regression","title":"Variational Imbalanced Regression: Fair Uncertainty Quantification via Probabilistic Smoothing","arxiv_id":"2306.06599","date":"2023-06-11","proceeding":"NeurIPS 2023 11","authors":["Ziyan Wang","Hao Wang"],"abstract":"Existing regression models tend to fall short in both accuracy and uncertainty estimation when the label distribution is imbalanced. In this paper, we propose a probabilistic deep learning model, dubbed variational imbalanced regression (VIR), which not only performs well in imbalanced regression but naturally produces reasonable uncertainty estimation as a byproduct. Different from typical variational autoencoders assuming I.I.D. representations (a data point's representation is not directly affected by other data points), our VIR borrows data with similar regression labels to compute the latent representation's variational distribution; furthermore, different from deterministic regression models producing point estimates, VIR predicts the entire normal-inverse-gamma distributions and modulates the associated conjugate distributions to impose probabilistic reweighting on the imbalanced data, thereby providing better uncertainty estimation. Experiments in several real-world datasets show that our VIR can outperform state-of-the-art imbalanced regression models in terms of both accuracy and uncertainty estimation. Code will soon be available at https://github.com/Wang-ML-Lab/variational-imbalanced-regression.","url_abs":"https://arxiv.org/abs/2306.06599v8","url_pdf":"https://arxiv.org/pdf/2306.06599v8.pdf","source":{"archive":"pwc-archive (Hugging Face), CC BY-SA 4.0","snapshot":"2025-07-28","licence_url":"https://creativecommons.org/licenses/by-sa/4.0/legalcode","row_kind":"abstracts"},"code_links":[{"paper_slug":"variational-imbalanced-regression","repo_url":"https://github.com/wang-ml-lab/variational-imbalanced-regression","is_official":1,"mentioned_in_paper":1,"mentioned_in_github":0,"framework":"pytorch","reach":{"status":"ok"}}],"tasks":[{"task_slug":"probabilistic-deep-learning","task_name":"Probabilistic Deep Learning"},{"task_slug":"uncertainty-quantification","task_name":"Uncertainty Quantification"},{"task_slug":"regression-1","task_name":"regression"}],"methods":[],"datasets_introduced":[],"methods_introduced":[],"results":[],"syntology":{"atlas_url":"https://app.syntology.ai/?focus=2306.06599","mcp":{"get_harvested_code_for_paper":{"arxiv_id":"2306.06599"}},"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":[{"provenance":"deterministic:regex_extraction","url":"https://github.com/Wang-ML-Lab/variational-imbalanced-regression","reach":{"status":"ok"}},{"provenance":"external:paperswithcode_snapshot_2025-07-28","url":"https://github.com/wang-ml-lab/variational-imbalanced-regression","reach":{"status":"ok"}}],"summary":{"ran_draft_wrong":6,"unverified":1},"by_repo_kind":{"official":{"samples":7,"ran":6,"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":7,"samples":[{"code_sha256_prefix":"cf81ba5eb189baed","entry":"EvidentialRegression","repo":"wang-ml-lab/variational-imbalanced-regression","repo_kind":"official","path":"evidential_deep_learning.py","file_url":"https://github.com/wang-ml-lab/variational-imbalanced-regression/blob/HEAD/evidential_deep_learning.py","link_basis":"first_harvest_node","language":"python","status":"ran_draft_wrong","verification_level":1,"contract_check":"MISDECLARED","metamorphic_tier":null,"behaviour_fingerprint":true,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"cf81ba5eb189baed"}},{"code_sha256_prefix":"4f9bc32a0ae6a2f4","entry":"KL_NIG","repo":"wang-ml-lab/variational-imbalanced-regression","repo_kind":"official","path":"evidential_deep_learning.py","file_url":"https://github.com/wang-ml-lab/variational-imbalanced-regression/blob/HEAD/evidential_deep_learning.py","link_basis":"first_harvest_node","language":"python","status":"ran_draft_wrong","verification_level":1,"contract_check":"MISDECLARED","metamorphic_tier":null,"behaviour_fingerprint":true,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"4f9bc32a0ae6a2f4"}},{"code_sha256_prefix":"ecddbf96db873aef","entry":"NIG_NLL","repo":"wang-ml-lab/variational-imbalanced-regression","repo_kind":"official","path":"evidential_deep_learning.py","file_url":"https://github.com/wang-ml-lab/variational-imbalanced-regression/blob/HEAD/evidential_deep_learning.py","link_basis":"first_harvest_node","language":"python","status":"ran_draft_wrong","verification_level":1,"contract_check":"MISDECLARED","metamorphic_tier":null,"behaviour_fingerprint":true,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"ecddbf96db873aef"}},{"code_sha256_prefix":"0b159f594568acfa","entry":"NIG_Reg","repo":"wang-ml-lab/variational-imbalanced-regression","repo_kind":"official","path":"evidential_deep_learning.py","file_url":"https://github.com/wang-ml-lab/variational-imbalanced-regression/blob/HEAD/evidential_deep_learning.py","link_basis":"first_harvest_node","language":"python","status":"ran_draft_wrong","verification_level":1,"contract_check":"MISDECLARED","metamorphic_tier":null,"behaviour_fingerprint":true,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"0b159f594568acfa"}},{"code_sha256_prefix":"444eb59da94fae25","entry":"calibrate_mean_var","repo":"wang-ml-lab/variational-imbalanced-regression","repo_kind":"official","path":"utils.py","file_url":"https://github.com/wang-ml-lab/variational-imbalanced-regression/blob/HEAD/utils.py","link_basis":"first_harvest_node","language":"python","status":"ran_draft_wrong","verification_level":1,"contract_check":"MISDECLARED","metamorphic_tier":"invariant","behaviour_fingerprint":true,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"444eb59da94fae25"}},{"code_sha256_prefix":"00eea8550969d77b","entry":"calibrate_mean_var_bayes","repo":"wang-ml-lab/variational-imbalanced-regression","repo_kind":"official","path":"utils.py","file_url":"https://github.com/wang-ml-lab/variational-imbalanced-regression/blob/HEAD/utils.py","link_basis":"first_harvest_node","language":"python","status":"ran_draft_wrong","verification_level":1,"contract_check":"MISDECLARED","metamorphic_tier":"invariant","behaviour_fingerprint":true,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"00eea8550969d77b"}},{"code_sha256_prefix":"06200c05f191b292","entry":"query_yes_no","repo":"wang-ml-lab/variational-imbalanced-regression","repo_kind":"official","path":"utils.py","file_url":"https://github.com/wang-ml-lab/variational-imbalanced-regression/blob/HEAD/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":"06200c05f191b292"}}]},"arxiv_metadata":null,"syntology_extracted_results":null}