{"id":"https://openalex.org/W4387838886","doi":"https://doi.org/10.48550/arxiv.2310.12680","title":"On the Optimization and Generalization of Multi-head Attention","display_name":"On the Optimization and Generalization of Multi-head Attention","publication_year":2023,"publication_date":"2023-10-19","ids":{"openalex":"https://openalex.org/W4387838886","doi":"https://doi.org/10.48550/arxiv.2310.12680"},"language":"en","primary_location":{"id":"pmh:oai:arXiv.org:2310.12680","is_oa":true,"landing_page_url":"http://arxiv.org/abs/2310.12680","pdf_url":"https://arxiv.org/pdf/2310.12680","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":"submittedVersion","is_accepted":false,"is_published":false,"raw_source_name":"","raw_type":"text"},"type":"preprint","indexed_in":["arxiv","datacite"],"open_access":{"is_oa":true,"oa_status":"green","oa_url":"https://arxiv.org/pdf/2310.12680","any_repository_has_fulltext":true},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5053095531","display_name":"Puneesh Deora","orcid":null},"institutions":[],"countries":[],"is_corresponding":true,"raw_author_name":"Deora, Puneesh","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5102570234","display_name":"Rouzbeh Ghaderi","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Ghaderi, Rouzbeh","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5102777812","display_name":"Hossein Taheri","orcid":"https://orcid.org/0000-0001-9597-995X"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Taheri, Hossein","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"last","author":{"id":"https://openalex.org/A5004173980","display_name":"Christos Thrampoulidis","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Thrampoulidis, Christos","raw_affiliation_strings":[],"affiliations":[]}],"institutions":[],"countries_distinct_count":0,"institutions_distinct_count":4,"corresponding_author_ids":["https://openalex.org/A5053095531"],"corresponding_institution_ids":[],"apc_list":null,"apc_paid":null,"fwci":null,"has_fulltext":true,"cited_by_count":4,"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/T10320","display_name":"Neural Networks and Applications","score":0.9969000220298767,"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/T10320","display_name":"Neural Networks and Applications","score":0.9969000220298767,"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/T11992","display_name":"CCD and CMOS Imaging Sensors","score":0.9962999820709229,"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/T11447","display_name":"Blind Source Separation Techniques","score":0.9932000041007996,"subfield":{"id":"https://openalex.org/subfields/1711","display_name":"Signal Processing"},"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/generalization","display_name":"Generalization","score":0.7948154807090759},{"id":"https://openalex.org/keywords/initialization","display_name":"Initialization","score":0.7622889280319214},{"id":"https://openalex.org/keywords/computer-science","display_name":"Computer science","score":0.7526007890701294},{"id":"https://openalex.org/keywords/realizability","display_name":"Realizability","score":0.6352365612983704},{"id":"https://openalex.org/keywords/artificial-intelligence","display_name":"Artificial intelligence","score":0.4623922109603882},{"id":"https://openalex.org/keywords/convergence","display_name":"Convergence (economics)","score":0.42293888330459595},{"id":"https://openalex.org/keywords/gradient-descent","display_name":"Gradient descent","score":0.4123355746269226},{"id":"https://openalex.org/keywords/mathematical-optimization","display_name":"Mathematical optimization","score":0.39056164026260376},{"id":"https://openalex.org/keywords/theoretical-computer-science","display_name":"Theoretical computer science","score":0.3819660544395447},{"id":"https://openalex.org/keywords/machine-learning","display_name":"Machine learning","score":0.33135277032852173},{"id":"https://openalex.org/keywords/algorithm","display_name":"Algorithm","score":0.3075895309448242},{"id":"https://openalex.org/keywords/artificial-neural-network","display_name":"Artificial neural network","score":0.21309593319892883},{"id":"https://openalex.org/keywords/mathematics","display_name":"Mathematics","score":0.16200214624404907}],"concepts":[{"id":"https://openalex.org/C177148314","wikidata":"https://www.wikidata.org/wiki/Q170084","display_name":"Generalization","level":2,"score":0.7948154807090759},{"id":"https://openalex.org/C114466953","wikidata":"https://www.wikidata.org/wiki/Q6034165","display_name":"Initialization","level":2,"score":0.7622889280319214},{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.7526007890701294},{"id":"https://openalex.org/C2776378722","wikidata":"https://www.wikidata.org/wiki/Q3454417","display_name":"Realizability","level":2,"score":0.6352365612983704},{"id":"https://openalex.org/C154945302","wikidata":"https://www.wikidata.org/wiki/Q11660","display_name":"Artificial intelligence","level":1,"score":0.4623922109603882},{"id":"https://openalex.org/C2777303404","wikidata":"https://www.wikidata.org/wiki/Q759757","display_name":"Convergence (economics)","level":2,"score":0.42293888330459595},{"id":"https://openalex.org/C153258448","wikidata":"https://www.wikidata.org/wiki/Q1199743","display_name":"Gradient descent","level":3,"score":0.4123355746269226},{"id":"https://openalex.org/C126255220","wikidata":"https://www.wikidata.org/wiki/Q141495","display_name":"Mathematical optimization","level":1,"score":0.39056164026260376},{"id":"https://openalex.org/C80444323","wikidata":"https://www.wikidata.org/wiki/Q2878974","display_name":"Theoretical computer science","level":1,"score":0.3819660544395447},{"id":"https://openalex.org/C119857082","wikidata":"https://www.wikidata.org/wiki/Q2539","display_name":"Machine learning","level":1,"score":0.33135277032852173},{"id":"https://openalex.org/C11413529","wikidata":"https://www.wikidata.org/wiki/Q8366","display_name":"Algorithm","level":1,"score":0.3075895309448242},{"id":"https://openalex.org/C50644808","wikidata":"https://www.wikidata.org/wiki/Q192776","display_name":"Artificial neural network","level":2,"score":0.21309593319892883},{"id":"https://openalex.org/C33923547","wikidata":"https://www.wikidata.org/wiki/Q395","display_name":"Mathematics","level":0,"score":0.16200214624404907},{"id":"https://openalex.org/C134306372","wikidata":"https://www.wikidata.org/wiki/Q7754","display_name":"Mathematical analysis","level":1,"score":0.0},{"id":"https://openalex.org/C199360897","wikidata":"https://www.wikidata.org/wiki/Q9143","display_name":"Programming language","level":1,"score":0.0},{"id":"https://openalex.org/C162324750","wikidata":"https://www.wikidata.org/wiki/Q8134","display_name":"Economics","level":0,"score":0.0},{"id":"https://openalex.org/C50522688","wikidata":"https://www.wikidata.org/wiki/Q189833","display_name":"Economic growth","level":1,"score":0.0}],"mesh":[],"locations_count":2,"locations":[{"id":"pmh:oai:arXiv.org:2310.12680","is_oa":true,"landing_page_url":"http://arxiv.org/abs/2310.12680","pdf_url":"https://arxiv.org/pdf/2310.12680","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":"submittedVersion","is_accepted":false,"is_published":false,"raw_source_name":"","raw_type":"text"},{"id":"doi:10.48550/arxiv.2310.12680","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2310.12680","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:oai:arXiv.org:2310.12680","is_oa":true,"landing_page_url":"http://arxiv.org/abs/2310.12680","pdf_url":"https://arxiv.org/pdf/2310.12680","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":"submittedVersion","is_accepted":false,"is_published":false,"raw_source_name":"","raw_type":"text"},"sustainable_development_goals":[],"awards":[{"id":"https://openalex.org/G1597412403","display_name":null,"funder_award_id":"RGPIN-","funder_id":"https://openalex.org/F4320334593","funder_display_name":"Natural Sciences and Engineering Research Council of Canada"},{"id":"https://openalex.org/G202906059","display_name":null,"funder_award_id":"03677","funder_id":"https://openalex.org/F4320334593","funder_display_name":"Natural Sciences and Engineering Research Council of Canada"},{"id":"https://openalex.org/G2222638181","display_name":null,"funder_award_id":"RGPIN-2021-03","funder_id":"https://openalex.org/F4320334593","funder_display_name":"Natural Sciences and Engineering Research Council of Canada"},{"id":"https://openalex.org/G6221715925","display_name":null,"funder_award_id":"RGPIN","funder_id":"https://openalex.org/F4320334593","funder_display_name":"Natural Sciences and Engineering Research Council of Canada"},{"id":"https://openalex.org/G6639552142","display_name":null,"funder_award_id":"2009030","funder_id":"https://openalex.org/F4320306076","funder_display_name":"National Science Foundation"},{"id":"https://openalex.org/G7608224839","display_name":null,"funder_award_id":"2021-03677","funder_id":"https://openalex.org/F4320334593","funder_display_name":"Natural Sciences and Engineering Research Council of Canada"},{"id":"https://openalex.org/G7833312070","display_name":null,"funder_award_id":"CCF-2009030","funder_id":"https://openalex.org/F4320306076","funder_display_name":"National Science Foundation"},{"id":"https://openalex.org/G8105784103","display_name":null,"funder_award_id":"RGPIN-202","funder_id":"https://openalex.org/F4320334593","funder_display_name":"Natural Sciences and Engineering Research Council of Canada"}],"funders":[{"id":"https://openalex.org/F4320306076","display_name":"National Science Foundation","ror":"https://ror.org/021nxhr62"},{"id":"https://openalex.org/F4320309949","display_name":"Canadian Institute for Advanced Research","ror":"https://ror.org/01sdtdd95"},{"id":"https://openalex.org/F4320334593","display_name":"Natural Sciences and Engineering Research Council of Canada","ror":"https://ror.org/01h531d29"}],"has_content":{"grobid_xml":false,"pdf":true},"content_urls":{"pdf":"https://content.openalex.org/works/W4387838886.pdf"},"referenced_works_count":0,"referenced_works":[],"related_works":["https://openalex.org/W1603488308","https://openalex.org/W4298120007","https://openalex.org/W2136737082","https://openalex.org/W3173554384","https://openalex.org/W1517224694","https://openalex.org/W2321910655","https://openalex.org/W4226299596","https://openalex.org/W4309434778","https://openalex.org/W3214759249","https://openalex.org/W4283773090"],"abstract_inverted_index":{"The":[0],"training":[1,32,59],"and":[2,40,54,109],"generalization":[3,41,55],"dynamics":[4],"of":[5,29,43,60],"the":[6,11,26,37,72,80,101],"Transformer's":[7],"core":[8],"mechanism,":[9,13],"namely":[10],"Attention":[12],"remain":[14],"under-explored.":[15],"Besides,":[16],"existing":[17],"analyses":[18],"primarily":[19],"focus":[20],"on":[21,71,79],"single-head":[22],"attention.":[23],"Inspired":[24],"by":[25],"demonstrated":[27],"benefits":[28],"overparameterization":[30],"when":[31],"fully-connected":[33],"networks,":[34],"we":[35,51,87],"investigate":[36],"potential":[38],"optimization":[39],"advantages":[42],"using":[44],"multiple":[45],"attention":[46],"heads.":[47],"Towards":[48],"this":[49],"goal,":[50],"derive":[52],"convergence":[53],"guarantees":[56],"for":[57,94],"gradient-descent":[58],"a":[61,67,95],"single-layer":[62],"multi-head":[63],"self-attention":[64],"model,":[65],"under":[66],"suitable":[68],"realizability":[69,84],"condition":[70],"data.":[73],"We":[74,99],"then":[75],"establish":[76],"primitive":[77],"conditions":[78,91],"initialization":[81],"that":[82,89],"ensure":[83],"holds.":[85],"Finally,":[86],"demonstrate":[88],"these":[90],"are":[92],"satisfied":[93],"simple":[96],"tokenized-mixture":[97],"model.":[98],"expect":[100],"analysis":[102],"can":[103],"be":[104],"extended":[105],"to":[106],"various":[107],"data-model":[108],"architecture":[110],"variations.":[111]},"counts_by_year":[{"year":2025,"cited_by_count":4}],"updated_date":"2026-04-21T08:09:41.155169","created_date":"2025-10-10T00:00:00"}