{"id":"https://openalex.org/W7128720547","doi":"https://doi.org/10.48550/arxiv.2602.10346","title":"Geometry-Aware Decoding with Wasserstein-Regularized Truncation and Mass Penalties for Large Language Models","display_name":"Geometry-Aware Decoding with Wasserstein-Regularized Truncation and Mass Penalties for Large Language Models","publication_year":2026,"publication_date":"2026-02-10","ids":{"openalex":"https://openalex.org/W7128720547","doi":"https://doi.org/10.48550/arxiv.2602.10346"},"language":null,"primary_location":{"id":"pmh:doi:10.48550/arxiv.2602.10346","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":"publisher-specific-oa","license_id":"https://openalex.org/licenses/publisher-specific-oa","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/A5024347525","display_name":"Arash Gholami Davoodi","orcid":"https://orcid.org/0000-0001-8611-9229"},"institutions":[],"countries":[],"is_corresponding":true,"raw_author_name":"Davoodi, Arash Gholami","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5125763883","display_name":"Navid Rezazadeh","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Rezazadeh, Navid","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5114988003","display_name":"Seyed Pouyan Mousavi Davoudi","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Davoudi, Seyed Pouyan Mousavi","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"last","author":{"id":null,"display_name":"Pezeshkpour, Pouya","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Pezeshkpour, Pouya","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]}],"institutions":[],"countries_distinct_count":0,"institutions_distinct_count":4,"corresponding_author_ids":["https://openalex.org/A5024347525"],"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.34619998931884766,"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.34619998931884766,"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/T10181","display_name":"Natural Language Processing Techniques","score":0.16750000417232513,"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/T10775","display_name":"Generative Adversarial Networks and Image Synthesis","score":0.0658000037074089,"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"}}],"keywords":[{"id":"https://openalex.org/keywords/decoding-methods","display_name":"Decoding methods","score":0.7545999884605408},{"id":"https://openalex.org/keywords/security-token","display_name":"Security token","score":0.7257999777793884},{"id":"https://openalex.org/keywords/prefix","display_name":"Prefix","score":0.7016000151634216},{"id":"https://openalex.org/keywords/entropy","display_name":"Entropy (arrow of time)","score":0.5216000080108643},{"id":"https://openalex.org/keywords/probability-distribution","display_name":"Probability distribution","score":0.45899999141693115},{"id":"https://openalex.org/keywords/coherence","display_name":"Coherence (philosophical gambling strategy)","score":0.4456000030040741},{"id":"https://openalex.org/keywords/truncation","display_name":"Truncation (statistics)","score":0.39809998869895935},{"id":"https://openalex.org/keywords/language-model","display_name":"Language model","score":0.3961000144481659}],"concepts":[{"id":"https://openalex.org/C57273362","wikidata":"https://www.wikidata.org/wiki/Q576722","display_name":"Decoding methods","level":2,"score":0.7545999884605408},{"id":"https://openalex.org/C48145219","wikidata":"https://www.wikidata.org/wiki/Q1335365","display_name":"Security token","level":2,"score":0.7257999777793884},{"id":"https://openalex.org/C141603448","wikidata":"https://www.wikidata.org/wiki/Q134830","display_name":"Prefix","level":2,"score":0.7016000151634216},{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.5282999873161316},{"id":"https://openalex.org/C11413529","wikidata":"https://www.wikidata.org/wiki/Q8366","display_name":"Algorithm","level":1,"score":0.5241000056266785},{"id":"https://openalex.org/C106301342","wikidata":"https://www.wikidata.org/wiki/Q4117933","display_name":"Entropy (arrow of time)","level":2,"score":0.5216000080108643},{"id":"https://openalex.org/C149441793","wikidata":"https://www.wikidata.org/wiki/Q200726","display_name":"Probability distribution","level":2,"score":0.45899999141693115},{"id":"https://openalex.org/C2781181686","wikidata":"https://www.wikidata.org/wiki/Q4226068","display_name":"Coherence (philosophical gambling strategy)","level":2,"score":0.4456000030040741},{"id":"https://openalex.org/C80444323","wikidata":"https://www.wikidata.org/wiki/Q2878974","display_name":"Theoretical computer science","level":1,"score":0.43799999356269836},{"id":"https://openalex.org/C33923547","wikidata":"https://www.wikidata.org/wiki/Q395","display_name":"Mathematics","level":0,"score":0.4043000042438507},{"id":"https://openalex.org/C106195933","wikidata":"https://www.wikidata.org/wiki/Q7847935","display_name":"Truncation (statistics)","level":2,"score":0.39809998869895935},{"id":"https://openalex.org/C137293760","wikidata":"https://www.wikidata.org/wiki/Q3621696","display_name":"Language model","level":2,"score":0.3961000144481659},{"id":"https://openalex.org/C197096303","wikidata":"https://www.wikidata.org/wiki/Q869887","display_name":"Probability mass function","level":3,"score":0.3424000144004822},{"id":"https://openalex.org/C2780586882","wikidata":"https://www.wikidata.org/wiki/Q7520643","display_name":"Simple (philosophy)","level":2,"score":0.31630000472068787},{"id":"https://openalex.org/C52622258","wikidata":"https://www.wikidata.org/wiki/Q131222","display_name":"Information theory","level":2,"score":0.304500013589859},{"id":"https://openalex.org/C174348530","wikidata":"https://www.wikidata.org/wiki/Q188635","display_name":"Bridging (networking)","level":2,"score":0.302700012922287},{"id":"https://openalex.org/C48044578","wikidata":"https://www.wikidata.org/wiki/Q727490","display_name":"Scalability","level":2,"score":0.28189998865127563},{"id":"https://openalex.org/C2780069185","wikidata":"https://www.wikidata.org/wiki/Q7977945","display_name":"Equivalence (formal languages)","level":2,"score":0.2802000045776367},{"id":"https://openalex.org/C9679016","wikidata":"https://www.wikidata.org/wiki/Q1417473","display_name":"Principle of maximum entropy","level":2,"score":0.2800000011920929},{"id":"https://openalex.org/C151376022","wikidata":"https://www.wikidata.org/wiki/Q168698","display_name":"Exponential function","level":2,"score":0.265500009059906},{"id":"https://openalex.org/C193702766","wikidata":"https://www.wikidata.org/wiki/Q1414548","display_name":"Concurrency","level":2,"score":0.2635999917984009},{"id":"https://openalex.org/C1769480","wikidata":"https://www.wikidata.org/wiki/Q1345239","display_name":"Entropy encoding","level":3,"score":0.25380000472068787}],"mesh":[],"locations_count":2,"locations":[{"id":"pmh:doi:10.48550/arxiv.2602.10346","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":"publisher-specific-oa","license_id":"https://openalex.org/licenses/publisher-specific-oa","version":"submittedVersion","is_accepted":false,"is_published":false,"raw_source_name":null,"raw_type":"Article"},{"id":"doi:10.48550/arxiv.2602.10346","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2602.10346","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":null,"license_id":null,"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.10346","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":"publisher-specific-oa","license_id":"https://openalex.org/licenses/publisher-specific-oa","version":"submittedVersion","is_accepted":false,"is_published":false,"raw_source_name":null,"raw_type":"Article"},"sustainable_development_goals":[{"score":0.42462509870529175,"display_name":"Peace, Justice and strong institutions","id":"https://metadata.un.org/sdg/16"}],"awards":[],"funders":[],"has_content":{"grobid_xml":false,"pdf":false},"content_urls":null,"referenced_works_count":0,"referenced_works":[],"related_works":[],"abstract_inverted_index":{"Large":[0],"language":[1],"models":[2,154],"(LLMs)":[3],"must":[4],"balance":[5],"diversity":[6],"and":[7,28,126,149],"creativity":[8,182],"against":[9,66],"logical":[10],"coherence":[11],"in":[12],"open-ended":[13,185],"generation.":[14],"Existing":[15],"truncation-based":[16],"samplers":[17],"are":[18],"effective":[19],"but":[20,179],"largely":[21],"heuristic,":[22],"relying":[23],"mainly":[24],"on":[25,86,143],"probability":[26,64],"mass":[27,65],"entropy":[29,68],"while":[30,60],"ignoring":[31],"semantic":[32],"geometry":[33],"of":[34,69,103],"the":[35,53,58,67,70,81,87,90,101,136],"token":[36,98],"space.":[37],"We":[38,116],"present":[39],"Top-W,":[40],"a":[41,76,96,104,113],"geometry-aware":[42],"truncation":[43],"rule":[44],"that":[45,107,134,156,172],"uses":[46],"Wasserstein":[47],"distance-defined":[48],"over":[49],"token-embedding":[50],"geometry-to":[51],"keep":[52],"cropped":[54],"distribution":[55],"close":[56],"to":[57,95,166],"original,":[59],"explicitly":[61],"balancing":[62],"retained":[63],"kept":[71],"set.":[72],"Our":[73],"theory":[74],"yields":[75],"simple":[77],"closed-form":[78],"structure":[79],"for":[80],"fixed-potential":[82],"subset":[83],"update:":[84],"depending":[85],"mass-entropy":[88],"trade-off,":[89],"optimal":[91],"crop":[92],"either":[93],"collapses":[94],"single":[97],"or":[99,124],"takes":[100],"form":[102],"one-dimensional":[105],"prefix":[106],"can":[108],"be":[109],"found":[110],"efficiently":[111],"with":[112,129],"linear":[114],"scan.":[115],"implement":[117],"Top-W":[118,157,173],"using":[119],"efficient":[120],"geometry-based":[121],"potentials":[122],"(nearest-set":[123],"k-NN)":[125],"pair":[127],"it":[128],"an":[130],"alternating":[131],"decoding":[132,162],"routine":[133],"keeps":[135],"standard":[137],"truncation-and-sampling":[138],"interface":[139],"unchanged.":[140],"Extensive":[141],"experiments":[142],"four":[144],"benchmarks":[145],"(GSM8K,":[146],"GPQA,":[147],"AlpacaEval,":[148],"MT-Bench)":[150],"across":[151],"three":[152],"instruction-tuned":[153],"show":[155],"consistently":[158],"outperforms":[159],"prior":[160],"state-of-the-art":[161],"approaches":[163],"achieving":[164],"up":[165],"33.7%":[167],"improvement.":[168],"Moreover,":[169],"we":[170],"find":[171],"not":[174],"only":[175],"improves":[176],"accuracy-focused":[177],"performance,":[178],"also":[180],"boosts":[181],"under":[183],"judge-based":[184],"evaluation.":[186]},"counts_by_year":[],"updated_date":"2026-05-05T08:41:31.759640","created_date":"2026-02-13T00:00:00"}