{"id":"https://openalex.org/W7128622183","doi":"https://doi.org/10.48550/arxiv.2602.09501","title":"Where-to-Unmask: Ground-Truth-Guided Unmasking Order Learning for Masked Diffusion Language Models","display_name":"Where-to-Unmask: Ground-Truth-Guided Unmasking Order Learning for Masked Diffusion Language Models","publication_year":2026,"publication_date":"2026-02-10","ids":{"openalex":"https://openalex.org/W7128622183","doi":"https://doi.org/10.48550/arxiv.2602.09501"},"language":null,"primary_location":{"id":"pmh:doi:10.48550/arxiv.2602.09501","is_oa":true,"landing_page_url":null,"pdf_url":null,"source":{"id":"https://openalex.org/S4406922384","display_name":"Open MIND","issn_l":null,"issn":null,"is_oa":false,"is_in_doaj":false,"is_core":false,"host_organization":null,"host_organization_name":null,"host_organization_lineage":[],"host_organization_lineage_names":[],"type":"repository"},"license":"cc-by","license_id":"https://openalex.org/licenses/cc-by","version":"submittedVersion","is_accepted":false,"is_published":false,"raw_source_name":null,"raw_type":"Article"},"type":"preprint","indexed_in":["datacite"],"open_access":{"is_oa":true,"oa_status":"green","oa_url":null,"any_repository_has_fulltext":true},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5055248161","display_name":"Hikaru Asano","orcid":null},"institutions":[],"countries":[],"is_corresponding":true,"raw_author_name":"Asano, Hikaru","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5070075141","display_name":"Tadashi Kozuno","orcid":"https://orcid.org/0000-0002-8820-1362"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Kozuno, Tadashi","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5125630581","display_name":"Kuniaki Saito","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Saito, Kuniaki","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"last","author":{"id":"https://openalex.org/A5010710732","display_name":"Yukino Baba","orcid":"https://orcid.org/0000-0001-5310-9841"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Baba, Yukino","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]}],"institutions":[],"countries_distinct_count":0,"institutions_distinct_count":4,"corresponding_author_ids":["https://openalex.org/A5055248161"],"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/T10028","display_name":"Topic Modeling","score":0.40610000491142273,"subfield":{"id":"https://openalex.org/subfields/1702","display_name":"Artificial Intelligence"},"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/T10028","display_name":"Topic Modeling","score":0.40610000491142273,"subfield":{"id":"https://openalex.org/subfields/1702","display_name":"Artificial Intelligence"},"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/T11714","display_name":"Multimodal Machine Learning Applications","score":0.21930000185966492,"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/T11636","display_name":"Artificial Intelligence in Healthcare and Education","score":0.05389999970793724,"subfield":{"id":"https://openalex.org/subfields/2718","display_name":"Health Informatics"},"field":{"id":"https://openalex.org/fields/27","display_name":"Medicine"},"domain":{"id":"https://openalex.org/domains/4","display_name":"Health Sciences"}}],"keywords":[{"id":"https://openalex.org/keywords/oracle","display_name":"Oracle","score":0.7738999724388123},{"id":"https://openalex.org/keywords/security-token","display_name":"Security token","score":0.7706000208854675},{"id":"https://openalex.org/keywords/language-model","display_name":"Language model","score":0.5483999848365784},{"id":"https://openalex.org/keywords/heuristic","display_name":"Heuristic","score":0.5374000072479248},{"id":"https://openalex.org/keywords/margin","display_name":"Margin (machine learning)","score":0.460099995136261},{"id":"https://openalex.org/keywords/venn-diagram","display_name":"Venn diagram","score":0.43810001015663147},{"id":"https://openalex.org/keywords/inference","display_name":"Inference","score":0.37720000743865967},{"id":"https://openalex.org/keywords/reinforcement-learning","display_name":"Reinforcement learning","score":0.3589000105857849}],"concepts":[{"id":"https://openalex.org/C55166926","wikidata":"https://www.wikidata.org/wiki/Q2892946","display_name":"Oracle","level":2,"score":0.7738999724388123},{"id":"https://openalex.org/C48145219","wikidata":"https://www.wikidata.org/wiki/Q1335365","display_name":"Security token","level":2,"score":0.7706000208854675},{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.7343000173568726},{"id":"https://openalex.org/C154945302","wikidata":"https://www.wikidata.org/wiki/Q11660","display_name":"Artificial intelligence","level":1,"score":0.6054999828338623},{"id":"https://openalex.org/C137293760","wikidata":"https://www.wikidata.org/wiki/Q3621696","display_name":"Language model","level":2,"score":0.5483999848365784},{"id":"https://openalex.org/C173801870","wikidata":"https://www.wikidata.org/wiki/Q201413","display_name":"Heuristic","level":2,"score":0.5374000072479248},{"id":"https://openalex.org/C119857082","wikidata":"https://www.wikidata.org/wiki/Q2539","display_name":"Machine learning","level":1,"score":0.46399998664855957},{"id":"https://openalex.org/C774472","wikidata":"https://www.wikidata.org/wiki/Q6760393","display_name":"Margin (machine learning)","level":2,"score":0.460099995136261},{"id":"https://openalex.org/C55508974","wikidata":"https://www.wikidata.org/wiki/Q190763","display_name":"Venn diagram","level":2,"score":0.43810001015663147},{"id":"https://openalex.org/C204321447","wikidata":"https://www.wikidata.org/wiki/Q30642","display_name":"Natural language processing","level":1,"score":0.4050999879837036},{"id":"https://openalex.org/C2776214188","wikidata":"https://www.wikidata.org/wiki/Q408386","display_name":"Inference","level":2,"score":0.37720000743865967},{"id":"https://openalex.org/C97541855","wikidata":"https://www.wikidata.org/wiki/Q830687","display_name":"Reinforcement learning","level":2,"score":0.3589000105857849},{"id":"https://openalex.org/C77618280","wikidata":"https://www.wikidata.org/wiki/Q1155772","display_name":"Scheme (mathematics)","level":2,"score":0.3555999994277954},{"id":"https://openalex.org/C182306322","wikidata":"https://www.wikidata.org/wiki/Q1779371","display_name":"Order (exchange)","level":2,"score":0.3447999954223633},{"id":"https://openalex.org/C2776999362","wikidata":"https://www.wikidata.org/wiki/Q2349274","display_name":"Planner","level":2,"score":0.34459999203681946},{"id":"https://openalex.org/C140779682","wikidata":"https://www.wikidata.org/wiki/Q210868","display_name":"Sampling (signal processing)","level":3,"score":0.3407999873161316},{"id":"https://openalex.org/C90805587","wikidata":"https://www.wikidata.org/wiki/Q10944557","display_name":"Word (group theory)","level":2,"score":0.3124000132083893},{"id":"https://openalex.org/C193221554","wikidata":"https://www.wikidata.org/wiki/Q5153664","display_name":"Commonsense reasoning","level":2,"score":0.28769999742507935},{"id":"https://openalex.org/C2983448237","wikidata":"https://www.wikidata.org/wiki/Q1078276","display_name":"Language understanding","level":2,"score":0.27619999647140503},{"id":"https://openalex.org/C165064840","wikidata":"https://www.wikidata.org/wiki/Q1321061","display_name":"Matching (statistics)","level":2,"score":0.27160000801086426},{"id":"https://openalex.org/C22367795","wikidata":"https://www.wikidata.org/wiki/Q7625208","display_name":"Structured prediction","level":2,"score":0.26980000734329224},{"id":"https://openalex.org/C177264268","wikidata":"https://www.wikidata.org/wiki/Q1514741","display_name":"Set (abstract data type)","level":2,"score":0.251800000667572},{"id":"https://openalex.org/C195324797","wikidata":"https://www.wikidata.org/wiki/Q33742","display_name":"Natural language","level":2,"score":0.2502000033855438}],"mesh":[],"locations_count":2,"locations":[{"id":"pmh:doi:10.48550/arxiv.2602.09501","is_oa":true,"landing_page_url":null,"pdf_url":null,"source":{"id":"https://openalex.org/S4406922384","display_name":"Open MIND","issn_l":null,"issn":null,"is_oa":false,"is_in_doaj":false,"is_core":false,"host_organization":null,"host_organization_name":null,"host_organization_lineage":[],"host_organization_lineage_names":[],"type":"repository"},"license":"cc-by","license_id":"https://openalex.org/licenses/cc-by","version":"submittedVersion","is_accepted":false,"is_published":false,"raw_source_name":null,"raw_type":"Article"},{"id":"doi:10.48550/arxiv.2602.09501","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2602.09501","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":"article"}],"best_oa_location":{"id":"pmh:doi:10.48550/arxiv.2602.09501","is_oa":true,"landing_page_url":null,"pdf_url":null,"source":{"id":"https://openalex.org/S4406922384","display_name":"Open MIND","issn_l":null,"issn":null,"is_oa":false,"is_in_doaj":false,"is_core":false,"host_organization":null,"host_organization_name":null,"host_organization_lineage":[],"host_organization_lineage_names":[],"type":"repository"},"license":"cc-by","license_id":"https://openalex.org/licenses/cc-by","version":"submittedVersion","is_accepted":false,"is_published":false,"raw_source_name":null,"raw_type":"Article"},"sustainable_development_goals":[{"display_name":"Peace, Justice and strong institutions","score":0.7150485515594482,"id":"https://metadata.un.org/sdg/16"}],"awards":[],"funders":[],"has_content":{"pdf":false,"grobid_xml":false},"content_urls":null,"referenced_works_count":0,"referenced_works":[],"related_works":[],"abstract_inverted_index":{"Masked":[0],"Diffusion":[1],"Language":[2],"Models":[3],"(MDLMs)":[4],"generate":[5],"text":[6],"by":[7,46],"iteratively":[8],"filling":[9],"masked":[10,99,137],"tokens,":[11,71],"requiring":[12],"two":[13],"coupled":[14],"decisions":[15],"at":[16],"each":[17,97],"step:":[18],"which":[19,25],"positions":[20,94],"to":[21,27,131,147],"unmask":[22],"(where-to-unmask)":[23,42],"and":[24,81],"tokens":[26],"place":[28],"(what-to-unmask).":[29],"While":[30],"standard":[31,144],"MDLM":[32,145],"training":[33],"directly":[34],"optimizes":[35],"token":[36,80,156],"prediction":[37,157],"(what-to-unmask),":[38],"inference-time":[39],"unmasking":[40,89,107,127],"orders":[41],"are":[43],"typically":[44],"determined":[45],"heuristic":[47],"confidence":[48],"measures":[49],"or":[50],"trained":[51],"through":[52],"reinforcement":[53],"learning":[54],"with":[55],"costly":[56],"on-policy":[57],"rollouts.":[58],"To":[59],"address":[60],"this,":[61],"we":[62,123],"introduce":[63],"Gt-Margin,":[64],"a":[65,125],"position-wise":[66],"score":[67],"derived":[68],"from":[69,136],"ground-truth":[70],"defined":[72],"as":[73],"the":[74,78,133,155],"probability":[75],"margin":[76],"between":[77],"correct":[79],"its":[82],"strongest":[83],"alternative.":[84],"Gt-Margin":[85],"yields":[86],"an":[87],"oracle":[88,106,134],"order":[90,108],"that":[91,103],"prioritizes":[92],"easier":[93],"first":[95],"under":[96],"partially":[98],"state.":[100],"We":[101],"demonstrate":[102],"leveraging":[104],"this":[105,121],"significantly":[109],"enhances":[110],"final":[111],"generation":[112],"quality,":[113],"particularly":[114],"on":[115,120],"logical":[116],"reasoning":[117,151],"benchmarks.":[118],"Building":[119],"insight,":[122],"train":[124],"supervised":[126],"planner":[128,141],"via":[129],"learning-to-rank":[130],"imitate":[132],"ordering":[135],"contexts.":[138],"The":[139],"resulting":[140],"integrates":[142],"into":[143],"sampling":[146],"select":[148],"where-to-unmask,":[149],"improving":[150],"accuracy":[152],"without":[153],"modifying":[154],"model.":[158]},"counts_by_year":[],"updated_date":"2026-05-05T08:41:31.759640","created_date":"2026-02-12T00:00:00"}