{"id":"https://openalex.org/W7163422733","doi":"https://doi.org/10.48550/arxiv.2606.03347","title":"AugMask: Training Diffusion Models on Incomplete Tabular Data via Stochastic Augmentation and Masking","display_name":"AugMask: Training Diffusion Models on Incomplete Tabular Data via Stochastic Augmentation and Masking","publication_year":2026,"publication_date":"2026-06-02","ids":{"openalex":"https://openalex.org/W7163422733","doi":"https://doi.org/10.48550/arxiv.2606.03347"},"language":null,"primary_location":{"id":"doi:10.48550/arxiv.2606.03347","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2606.03347","pdf_url":null,"source":{"id":"https://openalex.org/S4306400194","display_name":"arXiv (Cornell University)","issn_l":null,"issn":null,"is_oa":true,"is_in_doaj":false,"is_core":false,"host_organization":"https://openalex.org/I205783295","host_organization_name":"Cornell University","host_organization_lineage":["https://openalex.org/I205783295"],"host_organization_lineage_names":[],"type":"repository"},"license":"cc-by","license_id":"https://openalex.org/licenses/cc-by","version":null,"is_accepted":false,"is_published":false,"raw_source_name":null,"raw_type":"Preprint"},"type":"preprint","indexed_in":["datacite"],"open_access":{"is_oa":true,"oa_status":"green","oa_url":"https://doi.org/10.48550/arxiv.2606.03347","any_repository_has_fulltext":true},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5055627567","display_name":"Jungkyu Kim","orcid":"https://orcid.org/0000-0003-3691-6953"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Kim, Jungkyu","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5015048868","display_name":"Taeyoung Park","orcid":"https://orcid.org/0000-0001-7405-0746"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Park, Taeyoung","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"last","author":{"id":"https://openalex.org/A5137789522","display_name":"Kibok Lee","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Lee, Kibok","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]}],"institutions":[],"countries_distinct_count":0,"institutions_distinct_count":0,"corresponding_author_ids":[],"corresponding_institution_ids":[],"apc_list":null,"apc_paid":null,"fwci":null,"has_fulltext":false,"cited_by_count":0,"citation_normalized_percentile":null,"cited_by_percentile_year":null,"biblio":{"volume":null,"issue":null,"first_page":null,"last_page":null},"is_retracted":false,"is_paratext":false,"is_xpac":false,"primary_topic":{"id":"https://openalex.org/T10775","display_name":"Generative Adversarial Networks and Image Synthesis","score":0.7609000205993652,"subfield":{"id":"https://openalex.org/subfields/1707","display_name":"Computer Vision and Pattern Recognition"},"field":{"id":"https://openalex.org/fields/17","display_name":"Computer Science"},"domain":{"id":"https://openalex.org/domains/3","display_name":"Physical Sciences"}},"topics":[{"id":"https://openalex.org/T10775","display_name":"Generative Adversarial Networks and Image Synthesis","score":0.7609000205993652,"subfield":{"id":"https://openalex.org/subfields/1707","display_name":"Computer Vision and Pattern Recognition"},"field":{"id":"https://openalex.org/fields/17","display_name":"Computer Science"},"domain":{"id":"https://openalex.org/domains/3","display_name":"Physical Sciences"}},{"id":"https://openalex.org/T11206","display_name":"Model Reduction and Neural Networks","score":0.05350000038743019,"subfield":{"id":"https://openalex.org/subfields/3109","display_name":"Statistical and Nonlinear Physics"},"field":{"id":"https://openalex.org/fields/31","display_name":"Physics and Astronomy"},"domain":{"id":"https://openalex.org/domains/3","display_name":"Physical Sciences"}},{"id":"https://openalex.org/T12303","display_name":"Tensor decomposition and applications","score":0.022299999371170998,"subfield":{"id":"https://openalex.org/subfields/2605","display_name":"Computational Mathematics"},"field":{"id":"https://openalex.org/fields/26","display_name":"Mathematics"},"domain":{"id":"https://openalex.org/domains/3","display_name":"Physical Sciences"}}],"keywords":[{"id":"https://openalex.org/keywords/missing-data","display_name":"Missing data","score":0.8300999999046326},{"id":"https://openalex.org/keywords/context","display_name":"Context (archaeology)","score":0.6259999871253967},{"id":"https://openalex.org/keywords/training-set","display_name":"Training set","score":0.5572999715805054},{"id":"https://openalex.org/keywords/masking","display_name":"Masking (illustration)","score":0.4932999908924103},{"id":"https://openalex.org/keywords/training","display_name":"Training (meteorology)","score":0.4438000023365021},{"id":"https://openalex.org/keywords/generative-model","display_name":"Generative model","score":0.4408000111579895},{"id":"https://openalex.org/keywords/generative-grammar","display_name":"Generative grammar","score":0.4056999981403351},{"id":"https://openalex.org/keywords/synthetic-data","display_name":"Synthetic data","score":0.37229999899864197}],"concepts":[{"id":"https://openalex.org/C9357733","wikidata":"https://www.wikidata.org/wiki/Q6878417","display_name":"Missing data","level":2,"score":0.8300999999046326},{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.6793000102043152},{"id":"https://openalex.org/C2779343474","wikidata":"https://www.wikidata.org/wiki/Q3109175","display_name":"Context (archaeology)","level":2,"score":0.6259999871253967},{"id":"https://openalex.org/C51632099","wikidata":"https://www.wikidata.org/wiki/Q3985153","display_name":"Training set","level":2,"score":0.5572999715805054},{"id":"https://openalex.org/C154945302","wikidata":"https://www.wikidata.org/wiki/Q11660","display_name":"Artificial intelligence","level":1,"score":0.5152000188827515},{"id":"https://openalex.org/C2777402240","wikidata":"https://www.wikidata.org/wiki/Q6783436","display_name":"Masking (illustration)","level":2,"score":0.4932999908924103},{"id":"https://openalex.org/C124101348","wikidata":"https://www.wikidata.org/wiki/Q172491","display_name":"Data mining","level":1,"score":0.460999995470047},{"id":"https://openalex.org/C2777211547","wikidata":"https://www.wikidata.org/wiki/Q17141490","display_name":"Training (meteorology)","level":2,"score":0.4438000023365021},{"id":"https://openalex.org/C119857082","wikidata":"https://www.wikidata.org/wiki/Q2539","display_name":"Machine learning","level":1,"score":0.44339999556541443},{"id":"https://openalex.org/C167966045","wikidata":"https://www.wikidata.org/wiki/Q5532625","display_name":"Generative model","level":3,"score":0.4408000111579895},{"id":"https://openalex.org/C39890363","wikidata":"https://www.wikidata.org/wiki/Q36108","display_name":"Generative grammar","level":2,"score":0.4056999981403351},{"id":"https://openalex.org/C160920958","wikidata":"https://www.wikidata.org/wiki/Q7662746","display_name":"Synthetic data","level":2,"score":0.37229999899864197},{"id":"https://openalex.org/C99498987","wikidata":"https://www.wikidata.org/wiki/Q2210247","display_name":"Noise (video)","level":3,"score":0.3718999922275543},{"id":"https://openalex.org/C192209626","wikidata":"https://www.wikidata.org/wiki/Q190909","display_name":"Focus (optics)","level":2,"score":0.3650999963283539},{"id":"https://openalex.org/C11413529","wikidata":"https://www.wikidata.org/wiki/Q8366","display_name":"Algorithm","level":1,"score":0.34619998931884766},{"id":"https://openalex.org/C163294075","wikidata":"https://www.wikidata.org/wiki/Q581861","display_name":"Noise reduction","level":2,"score":0.30550000071525574},{"id":"https://openalex.org/C153180895","wikidata":"https://www.wikidata.org/wiki/Q7148389","display_name":"Pattern recognition (psychology)","level":2,"score":0.3041999936103821},{"id":"https://openalex.org/C2781170535","wikidata":"https://www.wikidata.org/wiki/Q30587856","display_name":"Noisy data","level":2,"score":0.28060001134872437},{"id":"https://openalex.org/C67186912","wikidata":"https://www.wikidata.org/wiki/Q367664","display_name":"Data modeling","level":2,"score":0.2768999934196472},{"id":"https://openalex.org/C44492722","wikidata":"https://www.wikidata.org/wiki/Q327069","display_name":"Conditional probability","level":2,"score":0.2745000123977661},{"id":"https://openalex.org/C69357855","wikidata":"https://www.wikidata.org/wiki/Q163214","display_name":"Diffusion","level":2,"score":0.26409998536109924}],"mesh":[],"locations_count":1,"locations":[{"id":"doi:10.48550/arxiv.2606.03347","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2606.03347","pdf_url":null,"source":{"id":"https://openalex.org/S4306400194","display_name":"arXiv (Cornell University)","issn_l":null,"issn":null,"is_oa":true,"is_in_doaj":false,"is_core":false,"host_organization":"https://openalex.org/I205783295","host_organization_name":"Cornell University","host_organization_lineage":["https://openalex.org/I205783295"],"host_organization_lineage_names":[],"type":"repository"},"license":"cc-by","license_id":"https://openalex.org/licenses/cc-by","version":null,"is_accepted":false,"is_published":null,"raw_source_name":null,"raw_type":"Preprint"}],"best_oa_location":{"id":"doi:10.48550/arxiv.2606.03347","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2606.03347","pdf_url":null,"source":{"id":"https://openalex.org/S4306400194","display_name":"arXiv (Cornell University)","issn_l":null,"issn":null,"is_oa":true,"is_in_doaj":false,"is_core":false,"host_organization":"https://openalex.org/I205783295","host_organization_name":"Cornell University","host_organization_lineage":["https://openalex.org/I205783295"],"host_organization_lineage_names":[],"type":"repository"},"license":"cc-by","license_id":"https://openalex.org/licenses/cc-by","version":null,"is_accepted":false,"is_published":false,"raw_source_name":null,"raw_type":"Preprint"},"sustainable_development_goals":[{"display_name":"Reduced inequalities","id":"https://metadata.un.org/sdg/10","score":0.6213266253471375}],"awards":[],"funders":[],"has_content":{"pdf":false,"grobid_xml":false},"content_urls":null,"referenced_works_count":0,"referenced_works":[],"related_works":[],"abstract_inverted_index":{"Score-based":[0],"diffusion":[1],"models":[2],"have":[3],"emerged":[4],"as":[5,80],"prominent":[6],"deep":[7],"generative":[8],"models;":[9],"however,":[10],"their":[11,19],"application":[12],"to":[13,44,71,93,125],"tabular":[14,27,123],"data":[15,28,46],"remains":[16],"challenging":[17],"because":[18],"backbones":[20,43],"assume":[21],"fully":[22],"specified":[23],"inputs,":[24],"whereas":[25],"real-world":[26],"often":[29],"contain":[30],"missing":[31,77,101],"values.":[32],"We":[33,88],"propose":[34],"AugMask,":[35],"a":[36,94,104],"plug-and-play":[37],"training":[38,86,91],"framework":[39],"that":[40,99],"adapts":[41],"missing-unaware":[42],"incomplete":[45],"by":[47],"separating":[48],"conditioning":[49,82],"from":[50],"supervision.":[51],"AugMask":[52,119],"1)":[53],"constructs":[54],"numeric":[55],"inputs":[56],"via":[57],"conditional":[58],"stochastic":[59],"augmentation":[60],"using":[61],"lightweight":[62],"auxiliary":[63],"models,":[64],"and":[65,97,116],"2)":[66],"applies":[67],"denoising":[68],"supervision":[69],"only":[70],"observed":[72],"coordinates.":[73],"In":[74],"effect,":[75],"augmented":[76],"entries":[78,102],"serve":[79],"uncertain":[81,111],"context":[83],"rather":[84],"than":[85],"targets.":[87],"connect":[89],"this":[90],"rule":[92],"Rao--Blackwellized":[95],"objective":[96],"show":[98],"marginalizing":[100],"yields":[103],"variance-weighted":[105],"sensitivity":[106],"penalty,":[107],"discouraging":[108],"over-reliance":[109],"on":[110],"completions.":[112],"Across":[113],"diverse":[114],"datasets":[115],"missingness":[117],"regimes,":[118],"enables":[120],"standard":[121],"diffusion-based":[122],"generators":[124],"outperform":[126],"specialized":[127],"missing-aware":[128],"baselines.":[129]},"counts_by_year":[],"updated_date":"2026-07-01T06:00:48.157686","created_date":"2026-06-04T00:00:00"}