{"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/dirichlet-based-prediction-calibration-for","title":"Dirichlet-Based Prediction Calibration for Learning with Noisy Labels","arxiv_id":"2401.07062","date":"2024-01-13","proceeding":null,"authors":["Chen-Chen Zong","Ye-Wen Wang","Ming-Kun Xie","Sheng-Jun Huang"],"abstract":"Learning with noisy labels can significantly hinder the generalization performance of deep neural networks (DNNs). Existing approaches address this issue through loss correction or example selection methods. However, these methods often rely on the model's predictions obtained from the softmax function, which can be over-confident and unreliable. In this study, we identify the translation invariance of the softmax function as the underlying cause of this problem and propose the \\textit{Dirichlet-based Prediction Calibration} (DPC) method as a solution. Our method introduces a calibrated softmax function that breaks the translation invariance by incorporating a suitable constant in the exponent term, enabling more reliable model predictions. To ensure stable model training, we leverage a Dirichlet distribution to assign probabilities to predicted labels and introduce a novel evidence deep learning (EDL) loss. The proposed loss function encourages positive and sufficiently large logits for the given label, while penalizing negative and small logits for other labels, leading to more distinct logits and facilitating better example selection based on a large-margin criterion. Through extensive experiments on diverse benchmark datasets, we demonstrate that DPC achieves state-of-the-art performance. The code is available at https://github.com/chenchenzong/DPC.","url_abs":"https://arxiv.org/abs/2401.07062v1","url_pdf":"https://arxiv.org/pdf/2401.07062v1.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":"dirichlet-based-prediction-calibration-for","repo_url":"https://github.com/chenchenzong/dpc","is_official":1,"mentioned_in_paper":1,"mentioned_in_github":1,"framework":"pytorch","reach":{"status":"ok"}}],"tasks":[{"task_slug":"learning-with-noisy-labels","task_name":"Learning with noisy labels"},{"task_slug":"translation","task_name":"Translation"}],"methods":[{"method_slug":"softmax","method_name":"Softmax"}],"datasets_introduced":[],"methods_introduced":[],"results":[],"syntology":{"syntology_url":"https://syntology.ai/paper/2401.07062","atlas_url":"https://app.syntology.ai/?focus=2401.07062","mcp":{"get_harvested_code_for_paper":{"arxiv_id":"2401.07062"}},"developers":"https://syntology.ai/developers","read_at":"2026-09-25T09:33:49+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":"external:paperswithcode_snapshot_2025-07-28","url":"https://github.com/chenchenzong/dpc","reach":{"status":"ok"}},{"provenance":"deterministic:regex_extraction","url":"https://github.com/chenchenzong/DPC","reach":{"status":"ok"}}],"summary":{"ran":11,"ran_fixture":3,"ran_draft_wrong":1,"unverified":5},"by_repo_kind":{"official":{"samples":20,"ran":15,"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":20,"samples":[{"code_sha256_prefix":"d893855f4250b50e","entry":"EDL_Loss","repo":"chenchenzong/DPC","repo_kind":"official","path":"AAAI2024_DPC_code/loss.py","file_url":"https://github.com/chenchenzong/DPC/blob/HEAD/AAAI2024_DPC_code/loss.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":"d893855f4250b50e"}},{"code_sha256_prefix":"d825b77f356fed41","entry":"ResNet18","repo":"chenchenzong/dpc","repo_kind":"official","path":"AAAI2024_DPC_code/PreResNet.py","file_url":"https://github.com/chenchenzong/dpc/blob/HEAD/AAAI2024_DPC_code/PreResNet.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":"d825b77f356fed41"}},{"code_sha256_prefix":"da291a22483b4cb4","entry":"ResNet34","repo":"chenchenzong/dpc","repo_kind":"official","path":"AAAI2024_DPC_code/PreResNet.py","file_url":"https://github.com/chenchenzong/dpc/blob/HEAD/AAAI2024_DPC_code/PreResNet.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":"da291a22483b4cb4"}},{"code_sha256_prefix":"b92936d1f2ac3e6c","entry":"ShearX","repo":"chenchenzong/dpc","repo_kind":"official","path":"AAAI2024_DPC_code/augmentations.py","file_url":"https://github.com/chenchenzong/dpc/blob/HEAD/AAAI2024_DPC_code/augmentations.py","link_basis":"plan_row","language":"python","status":"ran_fixture","verification_level":1,"contract_check":"RAISES","metamorphic_tier":"well_formed","behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"b92936d1f2ac3e6c"}},{"code_sha256_prefix":"008c64b77c4ae4c9","entry":"ShearY","repo":"chenchenzong/dpc","repo_kind":"official","path":"AAAI2024_DPC_code/augmentations.py","file_url":"https://github.com/chenchenzong/dpc/blob/HEAD/AAAI2024_DPC_code/augmentations.py","link_basis":"plan_row","language":"python","status":"ran_fixture","verification_level":1,"contract_check":"RAISES","metamorphic_tier":"well_formed","behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"008c64b77c4ae4c9"}},{"code_sha256_prefix":"ab50d5bbcf589623","entry":"TranslateX","repo":"chenchenzong/dpc","repo_kind":"official","path":"AAAI2024_DPC_code/augmentations.py","file_url":"https://github.com/chenchenzong/dpc/blob/HEAD/AAAI2024_DPC_code/augmentations.py","link_basis":"plan_row","language":"python","status":"ran_fixture","verification_level":1,"contract_check":"RAISES","metamorphic_tier":"well_formed","behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"ab50d5bbcf589623"}},{"code_sha256_prefix":"28750a9288820715","entry":"build_uniform_P","repo":"chenchenzong/dpc","repo_kind":"official","path":"AAAI2024_DPC_code/Asymmetric_Noise.py","file_url":"https://github.com/chenchenzong/dpc/blob/HEAD/AAAI2024_DPC_code/Asymmetric_Noise.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":"28750a9288820715"}},{"code_sha256_prefix":"583f9780bdd00a45","entry":"conv3x3","repo":"chenchenzong/dpc","repo_kind":"official","path":"AAAI2024_DPC_code/PreResNet.py","file_url":"https://github.com/chenchenzong/dpc/blob/HEAD/AAAI2024_DPC_code/PreResNet.py","link_basis":"harvester_set","language":"python","status":"ran_draft_wrong","verification_level":1,"contract_check":"OUTPUT_MISDECLARED","metamorphic_tier":"deterministic","behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"583f9780bdd00a45"}},{"code_sha256_prefix":"40d3c39a403ae1f7","entry":"float_parameter","repo":"chenchenzong/dpc","repo_kind":"official","path":"AAAI2024_DPC_code/augmentation_archive.py","file_url":"https://github.com/chenchenzong/dpc/blob/HEAD/AAAI2024_DPC_code/augmentation_archive.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":true,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"40d3c39a403ae1f7"}},{"code_sha256_prefix":"7900a3a6deef908b","entry":"int_parameter","repo":"chenchenzong/dpc","repo_kind":"official","path":"AAAI2024_DPC_code/augmentation_archive.py","file_url":"https://github.com/chenchenzong/dpc/blob/HEAD/AAAI2024_DPC_code/augmentation_archive.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":true,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"7900a3a6deef908b"}},{"code_sha256_prefix":"fc3891d70f5eaad6","entry":"rotate_with_fill","repo":"chenchenzong/dpc","repo_kind":"official","path":"AAAI2024_DPC_code/autoaugment.py","file_url":"https://github.com/chenchenzong/dpc/blob/HEAD/AAAI2024_DPC_code/autoaugment.py","link_basis":"harvester_set","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":"fc3891d70f5eaad6"}},{"code_sha256_prefix":"894d54a2cf2612d2","entry":"shearX","repo":"chenchenzong/dpc","repo_kind":"official","path":"AAAI2024_DPC_code/autoaugment.py","file_url":"https://github.com/chenchenzong/dpc/blob/HEAD/AAAI2024_DPC_code/autoaugment.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":"894d54a2cf2612d2"}},{"code_sha256_prefix":"ba6e11bd124c9bb1","entry":"shearY","repo":"chenchenzong/dpc","repo_kind":"official","path":"AAAI2024_DPC_code/autoaugment.py","file_url":"https://github.com/chenchenzong/dpc/blob/HEAD/AAAI2024_DPC_code/autoaugment.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":"ba6e11bd124c9bb1"}},{"code_sha256_prefix":"5b7355a3c842750a","entry":"unbiased_edge","repo":"chenchenzong/dpc","repo_kind":"official","path":"AAAI2024_DPC_code/Asymmetric_Noise.py","file_url":"https://github.com/chenchenzong/dpc/blob/HEAD/AAAI2024_DPC_code/Asymmetric_Noise.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":"5b7355a3c842750a"}},{"code_sha256_prefix":"1c64b6c09a6cccc5","entry":"unbiased_mean_op","repo":"chenchenzong/dpc","repo_kind":"official","path":"AAAI2024_DPC_code/Asymmetric_Noise.py","file_url":"https://github.com/chenchenzong/dpc/blob/HEAD/AAAI2024_DPC_code/Asymmetric_Noise.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":"1c64b6c09a6cccc5"}},{"code_sha256_prefix":"bb48f83128c0e944","entry":"consistency_loss","repo":"chenchenzong/dpc","repo_kind":"official","path":"AAAI2024_DPC_code/Train_cifar_aug.py","file_url":"https://github.com/chenchenzong/dpc/blob/HEAD/AAAI2024_DPC_code/Train_cifar_aug.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":"bb48f83128c0e944"}},{"code_sha256_prefix":"99bdd51751eccda4","entry":"conv_p","repo":"chenchenzong/dpc","repo_kind":"official","path":"AAAI2024_DPC_code/Train_cifar.py","file_url":"https://github.com/chenchenzong/dpc/blob/HEAD/AAAI2024_DPC_code/Train_cifar.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":"99bdd51751eccda4"}},{"code_sha256_prefix":"db4b97e8d6dd3fd3","entry":"create_model","repo":"chenchenzong/dpc","repo_kind":"official","path":"AAAI2024_DPC_code/Train_webvision_parallel_edl.py","file_url":"https://github.com/chenchenzong/dpc/blob/HEAD/AAAI2024_DPC_code/Train_webvision_parallel_edl.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":"db4b97e8d6dd3fd3"}},{"code_sha256_prefix":"1e7c011c7c168d10","entry":"eval_train","repo":"chenchenzong/dpc","repo_kind":"official","path":"AAAI2024_DPC_code/Train_cifar.py","file_url":"https://github.com/chenchenzong/dpc/blob/HEAD/AAAI2024_DPC_code/Train_cifar.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":"1e7c011c7c168d10"}},{"code_sha256_prefix":"767f49a34c3cb79a","entry":"linear_rampup","repo":"chenchenzong/dpc","repo_kind":"official","path":"AAAI2024_DPC_code/Train_cifar.py","file_url":"https://github.com/chenchenzong/dpc/blob/HEAD/AAAI2024_DPC_code/Train_cifar.py","link_basis":"plan_row","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":"767f49a34c3cb79a"}}]},"arxiv_metadata":null,"syntology_extracted_results":null}