{"id":"https://openalex.org/W7163662684","doi":"https://doi.org/10.48550/arxiv.2606.05484","title":"Learned Subspace Compression for Communication-Efficient Pipeline Parallelism","display_name":"Learned Subspace Compression for Communication-Efficient Pipeline Parallelism","publication_year":2026,"publication_date":"2026-06-03","ids":{"openalex":"https://openalex.org/W7163662684","doi":"https://doi.org/10.48550/arxiv.2606.05484"},"language":null,"primary_location":{"id":"doi:10.48550/arxiv.2606.05484","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2606.05484","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.2606.05484","any_repository_has_fulltext":true},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5114761084","display_name":"Paul Janson","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Janson, Paul","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5086553821","display_name":"Edouard Oyallon","orcid":"https://orcid.org/0000-0002-4826-7527"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Oyallon, Edouard","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"last","author":{"id":"https://openalex.org/A5137919334","display_name":"Eugene Belilovsky","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Belilovsky, Eugene","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/T12808","display_name":"Ferroelectric and Negative Capacitance Devices","score":0.2223999947309494,"subfield":{"id":"https://openalex.org/subfields/2208","display_name":"Electrical and Electronic Engineering"},"field":{"id":"https://openalex.org/fields/22","display_name":"Engineering"},"domain":{"id":"https://openalex.org/domains/3","display_name":"Physical Sciences"}},"topics":[{"id":"https://openalex.org/T12808","display_name":"Ferroelectric and Negative Capacitance Devices","score":0.2223999947309494,"subfield":{"id":"https://openalex.org/subfields/2208","display_name":"Electrical and Electronic Engineering"},"field":{"id":"https://openalex.org/fields/22","display_name":"Engineering"},"domain":{"id":"https://openalex.org/domains/3","display_name":"Physical Sciences"}},{"id":"https://openalex.org/T10036","display_name":"Advanced Neural Network Applications","score":0.1678999960422516,"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/T12611","display_name":"Neural Networks and Reservoir Computing","score":0.1145000010728836,"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"}}],"keywords":[{"id":"https://openalex.org/keywords/pipeline","display_name":"Pipeline (software)","score":0.7149999737739563},{"id":"https://openalex.org/keywords/subspace-topology","display_name":"Subspace topology","score":0.6639999747276306},{"id":"https://openalex.org/keywords/orthogonality","display_name":"Orthogonality","score":0.6380000114440918},{"id":"https://openalex.org/keywords/codebook","display_name":"Codebook","score":0.6377000212669373},{"id":"https://openalex.org/keywords/data-compression","display_name":"Data compression","score":0.565500020980835},{"id":"https://openalex.org/keywords/projection","display_name":"Projection (relational algebra)","score":0.5498999953269958},{"id":"https://openalex.org/keywords/bottleneck","display_name":"Bottleneck","score":0.5397999882698059},{"id":"https://openalex.org/keywords/residual","display_name":"Residual","score":0.4058000147342682},{"id":"https://openalex.org/keywords/image-compression","display_name":"Image compression","score":0.4000999927520752}],"concepts":[{"id":"https://openalex.org/C43521106","wikidata":"https://www.wikidata.org/wiki/Q2165493","display_name":"Pipeline (software)","level":2,"score":0.7149999737739563},{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.6995999813079834},{"id":"https://openalex.org/C32834561","wikidata":"https://www.wikidata.org/wiki/Q660730","display_name":"Subspace topology","level":2,"score":0.6639999747276306},{"id":"https://openalex.org/C17137986","wikidata":"https://www.wikidata.org/wiki/Q215067","display_name":"Orthogonality","level":2,"score":0.6380000114440918},{"id":"https://openalex.org/C127759330","wikidata":"https://www.wikidata.org/wiki/Q637416","display_name":"Codebook","level":2,"score":0.6377000212669373},{"id":"https://openalex.org/C11413529","wikidata":"https://www.wikidata.org/wiki/Q8366","display_name":"Algorithm","level":1,"score":0.5857999920845032},{"id":"https://openalex.org/C78548338","wikidata":"https://www.wikidata.org/wiki/Q2493","display_name":"Data compression","level":2,"score":0.565500020980835},{"id":"https://openalex.org/C57493831","wikidata":"https://www.wikidata.org/wiki/Q3134666","display_name":"Projection (relational algebra)","level":2,"score":0.5498999953269958},{"id":"https://openalex.org/C2780513914","wikidata":"https://www.wikidata.org/wiki/Q18210350","display_name":"Bottleneck","level":2,"score":0.5397999882698059},{"id":"https://openalex.org/C155512373","wikidata":"https://www.wikidata.org/wiki/Q287450","display_name":"Residual","level":2,"score":0.4058000147342682},{"id":"https://openalex.org/C13481523","wikidata":"https://www.wikidata.org/wiki/Q412438","display_name":"Image compression","level":4,"score":0.4000999927520752},{"id":"https://openalex.org/C2777121530","wikidata":"https://www.wikidata.org/wiki/Q7512739","display_name":"Signal subspace","level":4,"score":0.39250001311302185},{"id":"https://openalex.org/C28855332","wikidata":"https://www.wikidata.org/wiki/Q198099","display_name":"Quantization (signal processing)","level":2,"score":0.39070001244544983},{"id":"https://openalex.org/C612670","wikidata":"https://www.wikidata.org/wiki/Q7616373","display_name":"Stiefel manifold","level":2,"score":0.3736000061035156},{"id":"https://openalex.org/C180016635","wikidata":"https://www.wikidata.org/wiki/Q2712821","display_name":"Compression (physics)","level":2,"score":0.3375999927520752},{"id":"https://openalex.org/C94835093","wikidata":"https://www.wikidata.org/wiki/Q3113333","display_name":"Data compression ratio","level":5,"score":0.3310999870300293},{"id":"https://openalex.org/C57273362","wikidata":"https://www.wikidata.org/wiki/Q576722","display_name":"Decoding methods","level":2,"score":0.3246000111103058},{"id":"https://openalex.org/C80444323","wikidata":"https://www.wikidata.org/wiki/Q2878974","display_name":"Theoretical computer science","level":1,"score":0.32109999656677246},{"id":"https://openalex.org/C2778562939","wikidata":"https://www.wikidata.org/wiki/Q1298791","display_name":"Synchronization (alternating current)","level":3,"score":0.30720001459121704},{"id":"https://openalex.org/C529865628","wikidata":"https://www.wikidata.org/wiki/Q1790740","display_name":"Manifold (fluid mechanics)","level":2,"score":0.3061999976634979},{"id":"https://openalex.org/C154945302","wikidata":"https://www.wikidata.org/wiki/Q11660","display_name":"Artificial intelligence","level":1,"score":0.3012000024318695},{"id":"https://openalex.org/C74193536","wikidata":"https://www.wikidata.org/wiki/Q574844","display_name":"Kernel (algebra)","level":2,"score":0.2992999851703644},{"id":"https://openalex.org/C2778192920","wikidata":"https://www.wikidata.org/wiki/Q16874989","display_name":"Signal compression","level":4,"score":0.29499998688697815},{"id":"https://openalex.org/C81081738","wikidata":"https://www.wikidata.org/wiki/Q55542","display_name":"Lossless compression","level":3,"score":0.28439998626708984},{"id":"https://openalex.org/C164226766","wikidata":"https://www.wikidata.org/wiki/Q7293202","display_name":"Rank (graph theory)","level":2,"score":0.26669999957084656},{"id":"https://openalex.org/C173608175","wikidata":"https://www.wikidata.org/wiki/Q232661","display_name":"Parallel computing","level":1,"score":0.2542000114917755}],"mesh":[],"locations_count":1,"locations":[{"id":"doi:10.48550/arxiv.2606.05484","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2606.05484","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.2606.05484","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2606.05484","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":{"Pipeline":[0],"parallelism":[1],"enables":[2],"training":[3,82],"of":[4,52,78],"large":[5],"language":[6],"models":[7,183],"that":[8,95,149,162,177,191],"exceed":[9],"single-device":[10],"memory,":[11],"yet":[12],"inter-stage":[13,97],"activation":[14,153],"communication":[15,157],"becomes":[16],"the":[17,57,75,198],"dominant":[18],"bottleneck":[19],"when":[20],"trained":[21],"on":[22],"low-bandwidth":[23],"networks.":[24],"Recent":[25],"work":[26],"in":[27,43,215],"this":[28,40],"area":[29],"has":[30],"proposed":[31],"using":[32],"fixed":[33,115],"orthogonal":[34,102],"projections":[35],"to":[36,55,63,186,197,220],"compress":[37],"activations.":[38],"However,":[39],"still":[41],"results":[42],"a":[44,50,65,84,93,100,114,172,211],"significant":[45],"performance":[46,208,216],"degradation":[47,209],"and":[48,124,201],"requires":[49],"number":[51],"non-standard":[53],"adaptations":[54],"constrain":[56],"optimization.":[58],"A":[59],"natural":[60],"alternative":[61],"is":[62],"learn":[64],"low":[66],"rank":[67],"projection":[68,103,170],"for":[69,151],"each":[70,120],"pipeline":[71,121,200],"stage,":[72],"however":[73],"maintaining":[74],"necessary":[76],"orthogonality":[77],"these":[79],"projectors":[80],"during":[81],"remains":[83],"challenge.":[85],"We":[86,159],"present":[87],"Manifold":[88],"Aware":[89],"Projection":[90],"Learning":[91],"(MAPL),":[92],"method":[94],"treats":[96],"compression":[98,130,205,218],"as":[99],"learnable":[101],"under":[104],"explicit":[105],"Stiefel":[106],"manifold":[107],"(orthogonal":[108],"matrices)":[109],"constraints.":[110],"Rather":[111],"than":[112],"prescribing":[113],"global":[116],"subspace,":[117],"MAPL":[118,192],"lets":[119],"stage":[122,141],"discover":[123],"continuously":[125],"adapt":[126],"its":[127],"own":[128],"task-optimal":[129],"subspace":[131],"via":[132],"manifold-constrained":[133],"steepest":[134],"descent.":[135],"To":[136],"recover":[137],"token-specific":[138],"signals":[139],"at":[140],"boundaries,":[142],"we":[143,163,189],"introduce":[144],"per-stage":[145],"factorized":[146],"anchor":[147],"embeddings":[148],"allow":[150],"full-rank":[152],"reconstruction":[154],"with":[155,171,206,210],"negligible":[156],"overhead.":[158],"further":[160],"show":[161,190],"can":[164,193,202],"incorporate":[165],"residual":[166],"vector":[167],"quantization":[168],"after":[169],"streaming":[173],"codebook":[174],"synchronization":[175],"protocol":[176],"amortizes":[178],"dictionary":[179],"communication.":[180],"Across":[181],"LLaMA":[182],"from":[184],"150M":[185],"1B":[187],"parameters":[188],"be":[194],"easily":[195],"applied":[196],"existing":[199],"achieve":[203],"high":[204],"neglibile":[207],"drastically":[212],"improved":[213],"tradeoffs":[214],"vs.":[217],"compared":[219],"Subspace":[221],"Networks.":[222]},"counts_by_year":[],"updated_date":"2026-07-28T07:46:37.118299","created_date":"2026-06-06T00:00:00"}