{"id":"https://openalex.org/W7160862358","doi":"https://doi.org/10.48550/arxiv.2605.07914","title":"Flatness and Gradient Alignment Are Both Necessary: Spectral-Aware Gradient-Aligned Exploration for Multi-Distribution Learning","display_name":"Flatness and Gradient Alignment Are Both Necessary: Spectral-Aware Gradient-Aligned Exploration for Multi-Distribution Learning","publication_year":2026,"publication_date":"2026-05-08","ids":{"openalex":"https://openalex.org/W7160862358","doi":"https://doi.org/10.48550/arxiv.2605.07914"},"language":null,"primary_location":{"id":"doi:10.48550/arxiv.2605.07914","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2605.07914","pdf_url":null,"source":{"id":"https://openalex.org/S4306400194","display_name":"arXiv (Cornell University)","issn_l":"2331-8422","issn":["2331-8422"],"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.2605.07914","any_repository_has_fulltext":true},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5020813515","display_name":"Aristotelis Ballas","orcid":"https://orcid.org/0000-0003-1683-8433"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Ballas, Aristotelis","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"last","author":{"id":"https://openalex.org/A5135902435","display_name":"Christos Diou","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Diou, Christos","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/T11307","display_name":"Domain Adaptation and Few-Shot Learning","score":0.45260000228881836,"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/T11307","display_name":"Domain Adaptation and Few-Shot Learning","score":0.45260000228881836,"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/T11612","display_name":"Stochastic Gradient Optimization Techniques","score":0.25,"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/T10057","display_name":"Face and Expression Recognition","score":0.04619999974966049,"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/hessian-matrix","display_name":"Hessian matrix","score":0.7075999975204468},{"id":"https://openalex.org/keywords/gradient-descent","display_name":"Gradient descent","score":0.6679999828338623},{"id":"https://openalex.org/keywords/curvature","display_name":"Curvature","score":0.609499990940094},{"id":"https://openalex.org/keywords/covariance","display_name":"Covariance","score":0.506600022315979},{"id":"https://openalex.org/keywords/flatness","display_name":"Flatness (cosmology)","score":0.4634999930858612},{"id":"https://openalex.org/keywords/perturbation","display_name":"Perturbation (astronomy)","score":0.39079999923706055},{"id":"https://openalex.org/keywords/noise","display_name":"Noise (video)","score":0.3626999855041504},{"id":"https://openalex.org/keywords/isotropy","display_name":"Isotropy","score":0.3621000051498413},{"id":"https://openalex.org/keywords/covariance-matrix","display_name":"Covariance matrix","score":0.3547999858856201}],"concepts":[{"id":"https://openalex.org/C203616005","wikidata":"https://www.wikidata.org/wiki/Q620495","display_name":"Hessian matrix","level":2,"score":0.7075999975204468},{"id":"https://openalex.org/C153258448","wikidata":"https://www.wikidata.org/wiki/Q1199743","display_name":"Gradient descent","level":3,"score":0.6679999828338623},{"id":"https://openalex.org/C195065555","wikidata":"https://www.wikidata.org/wiki/Q214881","display_name":"Curvature","level":2,"score":0.609499990940094},{"id":"https://openalex.org/C33923547","wikidata":"https://www.wikidata.org/wiki/Q395","display_name":"Mathematics","level":0,"score":0.536300003528595},{"id":"https://openalex.org/C178650346","wikidata":"https://www.wikidata.org/wiki/Q201984","display_name":"Covariance","level":2,"score":0.506600022315979},{"id":"https://openalex.org/C2778530986","wikidata":"https://www.wikidata.org/wiki/Q5457948","display_name":"Flatness (cosmology)","level":3,"score":0.4634999930858612},{"id":"https://openalex.org/C11413529","wikidata":"https://www.wikidata.org/wiki/Q8366","display_name":"Algorithm","level":1,"score":0.4284000098705292},{"id":"https://openalex.org/C177918212","wikidata":"https://www.wikidata.org/wiki/Q803623","display_name":"Perturbation (astronomy)","level":2,"score":0.39079999923706055},{"id":"https://openalex.org/C99498987","wikidata":"https://www.wikidata.org/wiki/Q2210247","display_name":"Noise (video)","level":3,"score":0.3626999855041504},{"id":"https://openalex.org/C184050105","wikidata":"https://www.wikidata.org/wiki/Q273163","display_name":"Isotropy","level":2,"score":0.3621000051498413},{"id":"https://openalex.org/C185142706","wikidata":"https://www.wikidata.org/wiki/Q1134404","display_name":"Covariance matrix","level":2,"score":0.3547999858856201},{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.34470000863075256},{"id":"https://openalex.org/C189950617","wikidata":"https://www.wikidata.org/wiki/Q937228","display_name":"Property (philosophy)","level":2,"score":0.34369999170303345},{"id":"https://openalex.org/C61797465","wikidata":"https://www.wikidata.org/wiki/Q1188986","display_name":"Term (time)","level":2,"score":0.33660000562667847},{"id":"https://openalex.org/C134306372","wikidata":"https://www.wikidata.org/wiki/Q7754","display_name":"Mathematical analysis","level":1,"score":0.33230000734329224},{"id":"https://openalex.org/C115680565","wikidata":"https://www.wikidata.org/wiki/Q5977448","display_name":"Gradient method","level":2,"score":0.33070001006126404},{"id":"https://openalex.org/C168167062","wikidata":"https://www.wikidata.org/wiki/Q1117970","display_name":"Component (thermodynamics)","level":2,"score":0.3271999955177307},{"id":"https://openalex.org/C184720557","wikidata":"https://www.wikidata.org/wiki/Q7825049","display_name":"Topology (electrical circuits)","level":2,"score":0.323199987411499},{"id":"https://openalex.org/C27438332","wikidata":"https://www.wikidata.org/wiki/Q2873","display_name":"Principal component analysis","level":2,"score":0.3206999897956848},{"id":"https://openalex.org/C32834561","wikidata":"https://www.wikidata.org/wiki/Q660730","display_name":"Subspace topology","level":2,"score":0.310699999332428},{"id":"https://openalex.org/C47446073","wikidata":"https://www.wikidata.org/wiki/Q5165890","display_name":"Control theory (sociology)","level":3,"score":0.2709999978542328},{"id":"https://openalex.org/C57691317","wikidata":"https://www.wikidata.org/wiki/Q1289248","display_name":"Scalar (mathematics)","level":2,"score":0.2660999894142151},{"id":"https://openalex.org/C2524010","wikidata":"https://www.wikidata.org/wiki/Q8087","display_name":"Geometry","level":1,"score":0.2651999890804291},{"id":"https://openalex.org/C112972136","wikidata":"https://www.wikidata.org/wiki/Q7595718","display_name":"Stability (learning theory)","level":2,"score":0.26429998874664307},{"id":"https://openalex.org/C28826006","wikidata":"https://www.wikidata.org/wiki/Q33521","display_name":"Applied mathematics","level":1,"score":0.2597000002861023},{"id":"https://openalex.org/C122748992","wikidata":"https://www.wikidata.org/wiki/Q2029312","display_name":"Gradient analysis","level":3,"score":0.2574999928474426},{"id":"https://openalex.org/C51432778","wikidata":"https://www.wikidata.org/wiki/Q1259145","display_name":"Independent component analysis","level":2,"score":0.2549999952316284}],"mesh":[],"locations_count":1,"locations":[{"id":"doi:10.48550/arxiv.2605.07914","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2605.07914","pdf_url":null,"source":{"id":"https://openalex.org/S4306400194","display_name":"arXiv (Cornell University)","issn_l":"2331-8422","issn":["2331-8422"],"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.2605.07914","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2605.07914","pdf_url":null,"source":{"id":"https://openalex.org/S4306400194","display_name":"arXiv (Cornell University)","issn_l":"2331-8422","issn":["2331-8422"],"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":[],"awards":[],"funders":[],"has_content":{"pdf":false,"grobid_xml":false},"content_urls":null,"referenced_works_count":0,"referenced_works":[],"related_works":[],"abstract_inverted_index":{"Sharpness-aware":[0],"and":[1,39,43,75,89,105,208,224],"gradient-alignment":[2],"methods":[3,14],"have":[4],"been":[5],"shown":[6],"to":[7,230],"improve":[8],"generalization,":[9],"however":[10],"each":[11,162],"family":[12],"of":[13,20,73,94,161,197],"targets":[15,147],"a":[16,77,114,219,227],"single":[17],"geometric":[18],"property":[19],"the":[21,26,71,86,92,95,108,120,158,172,182,185,192,195,215],"loss":[22],"landscape,":[23],"while":[24],"ignoring":[25],"other.":[27,109],"In":[28],"this":[29,34,138],"paper,":[30],"we":[31,54,140],"show":[32,213],"that":[33,40,59,116,146,171,214],"omission":[35],"is":[36,85,91],"structurally":[37],"unavoidable":[38],"both":[41,148],"flatness":[42],"gradient":[44,96,164,202],"alignment":[45,67,186],"should":[46],"be":[47],"considered":[48],"in":[49,103,107,122],"multi-distribution":[50],"learning":[51,211],"settings.":[52],"Specifically,":[53],"derive":[55],"an":[56,66],"excess-risk":[57],"decomposition":[58],"yields":[60],"two":[61,209],"additive":[62],"leading-order":[63],"terms:":[64],"(i)":[65],"term,":[68,79],"controlled":[69,80],"by":[70,81,137],"trace":[72],"$\\bar{H}^{-1}\u03a3_g$":[74],"(ii)":[76],"curvature":[78,151],"$\\bar{H}$,":[82],"where":[83],"$\\bar{H}$":[84,100],"average":[87],"Hessian":[88],"$\u03a3_g$":[90],"covariance":[93],"across":[97],"distributions.":[98],"Notably,":[99],"appears":[101],"inverted":[102],"one":[104,129],"non-inverted":[106],"We":[110],"further":[111],"show,":[112],"via":[113,167],"counterexample,":[115],"neither":[117],"quantity":[118],"bounds":[119],"other":[121,183],"general,":[123],"so":[124,170],"no":[125],"algorithm":[126],"targeting":[127],"only":[128],"term":[130],"can":[131],"guarantee":[132],"low":[133],"excess":[134],"risk.":[135],"Motivated":[136],"decomposition,":[139],"propose":[141],"SAGE":[142],"(Spectral-Aware":[143],"Gradient-Aligned":[144],"Exploration)":[145],"terms.":[149],"The":[150],"component":[152,187],"replaces":[153],"SAM's":[154],"gradient-scaled":[155],"perturbation":[156],"with":[157,178,200],"polar":[159],"factor":[160],"layer's":[163],"matrix,":[165],"computed":[166],"Newton-Schulz":[168],"iteration,":[169],"ascent":[173],"step":[174],"probes":[175],"all":[176],"directions":[177],"similar":[179],"magnitude.":[180],"On":[181],"hand,":[184],"injects":[188],"isotropic":[189],"noise":[190],"at":[191],"descent":[193],"step,":[194],"magnitude":[196],"which":[198],"scales":[199],"cross-distribution":[201],"disagreement.":[203],"Experiments":[204],"on":[205,222],"five":[206],"domain-generalization":[207],"multi-task":[210],"benchmarks":[212],"proposed":[216],"method":[217],"establishes":[218],"new":[220],"state-of-the-art":[221,240],"DomainBed":[223],"acts":[225],"as":[226],"general-purpose":[228],"improvement":[229],"base":[231],"MTL":[232],"solvers,":[233],"remaining":[234],"competitive":[235],"with,":[236],"or":[237],"even":[238],"surpassing,":[239],"methods.":[241]},"counts_by_year":[],"updated_date":"2026-07-28T07:46:37.118299","created_date":"2026-05-12T00:00:00"}
