{"id":"https://openalex.org/W7162489486","doi":"https://doi.org/10.48550/arxiv.2605.26327","title":"Reparametrizing Shampoo and SOAP for Subspace Basis Updates and BFloat16 Storage","display_name":"Reparametrizing Shampoo and SOAP for Subspace Basis Updates and BFloat16 Storage","publication_year":2026,"publication_date":"2026-05-25","ids":{"openalex":"https://openalex.org/W7162489486","doi":"https://doi.org/10.48550/arxiv.2605.26327"},"language":null,"primary_location":{"id":"doi:10.48550/arxiv.2605.26327","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2605.26327","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":false,"raw_source_name":null,"raw_type":"article"},"type":"preprint","indexed_in":["datacite"],"open_access":{"is_oa":true,"oa_status":"green","oa_url":"https://doi.org/10.48550/arxiv.2605.26327","any_repository_has_fulltext":true},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5113617187","display_name":"Alan Milligan","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Milligan, Alan","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5137115433","display_name":"Zikun Xu","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Xu, Zikun","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5137117546","display_name":"Simon Lacoste-Julien","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Lacoste-Julien, Simon","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5012954635","display_name":"Felix Dangel","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Dangel, Felix","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"last","author":{"id":"https://openalex.org/A5137180162","display_name":"Wu Lin","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Lin, Wu","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]}],"institutions":[],"countries_distinct_count":0,"institutions_distinct_count":5,"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/T10054","display_name":"Parallel Computing and Optimization Techniques","score":0.3312000036239624,"subfield":{"id":"https://openalex.org/subfields/1708","display_name":"Hardware and Architecture"},"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/T10054","display_name":"Parallel Computing and Optimization Techniques","score":0.3312000036239624,"subfield":{"id":"https://openalex.org/subfields/1708","display_name":"Hardware and Architecture"},"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/T11697","display_name":"Numerical Methods and Algorithms","score":0.09160000085830688,"subfield":{"id":"https://openalex.org/subfields/1703","display_name":"Computational Theory and Mathematics"},"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/T10101","display_name":"Cloud Computing and Resource Management","score":0.06480000168085098,"subfield":{"id":"https://openalex.org/subfields/1710","display_name":"Information Systems"},"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/basis","display_name":"Basis (linear algebra)","score":0.6938999891281128},{"id":"https://openalex.org/keywords/overhead","display_name":"Overhead (engineering)","score":0.5458999872207642},{"id":"https://openalex.org/keywords/qr-decomposition","display_name":"QR decomposition","score":0.5425999760627747},{"id":"https://openalex.org/keywords/subspace-topology","display_name":"Subspace topology","score":0.46059998869895935},{"id":"https://openalex.org/keywords/implementation","display_name":"Implementation","score":0.4417000114917755},{"id":"https://openalex.org/keywords/decomposition","display_name":"Decomposition","score":0.4032000005245209},{"id":"https://openalex.org/keywords/artificial-neural-network","display_name":"Artificial neural network","score":0.37389999628067017},{"id":"https://openalex.org/keywords/matrix-decomposition","display_name":"Matrix decomposition","score":0.37209999561309814}],"concepts":[{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.7365999817848206},{"id":"https://openalex.org/C12426560","wikidata":"https://www.wikidata.org/wiki/Q189569","display_name":"Basis (linear algebra)","level":2,"score":0.6938999891281128},{"id":"https://openalex.org/C2779960059","wikidata":"https://www.wikidata.org/wiki/Q7113681","display_name":"Overhead (engineering)","level":2,"score":0.5458999872207642},{"id":"https://openalex.org/C188060507","wikidata":"https://www.wikidata.org/wiki/Q653242","display_name":"QR decomposition","level":3,"score":0.5425999760627747},{"id":"https://openalex.org/C32834561","wikidata":"https://www.wikidata.org/wiki/Q660730","display_name":"Subspace topology","level":2,"score":0.46059998869895935},{"id":"https://openalex.org/C26713055","wikidata":"https://www.wikidata.org/wiki/Q245962","display_name":"Implementation","level":2,"score":0.4417000114917755},{"id":"https://openalex.org/C124681953","wikidata":"https://www.wikidata.org/wiki/Q339062","display_name":"Decomposition","level":2,"score":0.4032000005245209},{"id":"https://openalex.org/C11413529","wikidata":"https://www.wikidata.org/wiki/Q8366","display_name":"Algorithm","level":1,"score":0.4016000032424927},{"id":"https://openalex.org/C50644808","wikidata":"https://www.wikidata.org/wiki/Q192776","display_name":"Artificial neural network","level":2,"score":0.37389999628067017},{"id":"https://openalex.org/C42355184","wikidata":"https://www.wikidata.org/wiki/Q1361088","display_name":"Matrix decomposition","level":3,"score":0.37209999561309814},{"id":"https://openalex.org/C17881449","wikidata":"https://www.wikidata.org/wiki/Q189620","display_name":"SOAP","level":2,"score":0.3594000041484833},{"id":"https://openalex.org/C2779679103","wikidata":"https://www.wikidata.org/wiki/Q5251805","display_name":"Degradation (telecommunications)","level":2,"score":0.30379998683929443},{"id":"https://openalex.org/C173608175","wikidata":"https://www.wikidata.org/wiki/Q232661","display_name":"Parallel computing","level":1,"score":0.29919999837875366},{"id":"https://openalex.org/C94375191","wikidata":"https://www.wikidata.org/wiki/Q11205","display_name":"Arithmetic","level":1,"score":0.2962999939918518},{"id":"https://openalex.org/C2775941552","wikidata":"https://www.wikidata.org/wiki/Q25212305","display_name":"Isolation (microbiology)","level":2,"score":0.28279998898506165},{"id":"https://openalex.org/C80444323","wikidata":"https://www.wikidata.org/wiki/Q2878974","display_name":"Theoretical computer science","level":1,"score":0.2802000045776367},{"id":"https://openalex.org/C5917680","wikidata":"https://www.wikidata.org/wiki/Q2621825","display_name":"Basis function","level":2,"score":0.27810001373291016},{"id":"https://openalex.org/C157764524","wikidata":"https://www.wikidata.org/wiki/Q1383412","display_name":"Throughput","level":3,"score":0.2736000120639801},{"id":"https://openalex.org/C113775141","wikidata":"https://www.wikidata.org/wiki/Q428691","display_name":"Computer engineering","level":1,"score":0.27090001106262207},{"id":"https://openalex.org/C2778258933","wikidata":"https://www.wikidata.org/wiki/Q16918986","display_name":"Decomposition method (queueing theory)","level":2,"score":0.2630000114440918},{"id":"https://openalex.org/C154945302","wikidata":"https://www.wikidata.org/wiki/Q11660","display_name":"Artificial intelligence","level":1,"score":0.2572000026702881},{"id":"https://openalex.org/C167431342","wikidata":"https://www.wikidata.org/wiki/Q1754327","display_name":"Preconditioner","level":3,"score":0.2547000050544739},{"id":"https://openalex.org/C2778112365","wikidata":"https://www.wikidata.org/wiki/Q3511065","display_name":"Sequence (biology)","level":2,"score":0.25279998779296875},{"id":"https://openalex.org/C56372850","wikidata":"https://www.wikidata.org/wiki/Q1050404","display_name":"Sparse matrix","level":3,"score":0.2517000138759613},{"id":"https://openalex.org/C77246614","wikidata":"https://www.wikidata.org/wiki/Q1409400","display_name":"Gramian matrix","level":3,"score":0.25130000710487366}],"mesh":[],"locations_count":1,"locations":[{"id":"doi:10.48550/arxiv.2605.26327","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2605.26327","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":"doi:10.48550/arxiv.2605.26327","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2605.26327","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":false,"raw_source_name":null,"raw_type":"article"},"sustainable_development_goals":[],"awards":[],"funders":[],"has_content":{"grobid_xml":false,"pdf":false},"content_urls":null,"referenced_works_count":0,"referenced_works":[],"related_works":[],"abstract_inverted_index":{"Shampoo-based":[0,58,116,151],"methods,":[1],"such":[2],"as":[3],"KL-Shampoo":[4],"and":[5,15,28,36,71,125,135,155],"SOAP,":[6,124],"have":[7],"demonstrated":[8],"strong":[9],"performance":[10,56,105,132],"in":[11,94],"training":[12],"neural":[13],"networks":[14],"rely":[16],"on":[17],"QR":[18,22,92,120],"decomposition.":[19],"Because":[20],"existing":[21],"implementations":[23],"require":[24],"single-precision":[25],"(FP32)":[26],"arithmetic":[27],"remain":[29],"computationally":[30],"expensive,":[31],"these":[32],"methods":[33,117,152],"become":[34],"time-":[35],"memory-intensive":[37],"when":[38],"their":[39],"preconditioning":[40],"matrices":[41],"are":[42],"large.":[43],"Moreover,":[44],"using":[45],"BFloat16":[46],"(BFP16)":[47],"storage":[48,70],"to":[49,115,142],"reduce":[50],"memory":[51],"usage":[52],"can":[53],"degrade":[54],"the":[55,65,89,104,131],"of":[57,64,88,133],"methods.":[59],"We":[60],"propose":[61],"a":[62,73,95],"reparametrization":[63],"preconditioner":[66],"that":[67,118],"supports":[68],"BFP16":[69,109,138],"forms":[72],"complete":[74],"basis":[75,79,90],"by":[76,108],"combining":[77],"updated":[78],"vectors":[80],"with":[81],"unchanged":[82],"ones.":[83],"By":[84],"updating":[85],"only":[86],"part":[87],"through":[91],"decomposition":[93],"subspace,":[96],"our":[97,148],"approach":[98,112,149],"reduces":[99],"computational":[100],"overhead":[101],"while":[102],"mitigating":[103],"degradation":[106],"caused":[107],"storage.":[110],"Our":[111],"applies":[113],"broadly":[114],"employ":[119],"decomposition,":[121],"including":[122],"KL-Shampoo,":[123],"KL-SOAP.":[126],"In":[127],"particular,":[128],"it":[129],"improves":[130],"SOAP":[134],"KL-SOAP":[136,141],"under":[137],"storage,":[139],"enabling":[140],"match":[143],"or":[144],"exceed":[145],"KL-Shampoo.":[146],"Overall,":[147],"makes":[150],"more":[153],"memory-":[154],"time-efficient.":[156]},"counts_by_year":[],"updated_date":"2026-06-11T09:08:48.828518","created_date":"2026-05-28T00:00:00"}