{"id":"https://openalex.org/W7131380569","doi":"https://doi.org/10.48550/arxiv.2602.20008","title":"Token-UNet: A New Case for Transformers Integration in Efficient and Interpretable 3D UNets for Brain Imaging Segmentation","display_name":"Token-UNet: A New Case for Transformers Integration in Efficient and Interpretable 3D UNets for Brain Imaging Segmentation","publication_year":2026,"publication_date":"2026-02-23","ids":{"openalex":"https://openalex.org/W7131380569","doi":"https://doi.org/10.48550/arxiv.2602.20008"},"language":null,"primary_location":{"id":"doi:10.48550/arxiv.2602.20008","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2602.20008","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":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.2602.20008","any_repository_has_fulltext":true},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5083475523","display_name":"Louis Fabrice Tshimanga","orcid":"https://orcid.org/0009-0002-1240-4830"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Tshimanga, Louis Fabrice","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5078442554","display_name":"Andrea Zanola","orcid":"https://orcid.org/0000-0001-6973-8634"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Zanola, Andrea","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5051115235","display_name":"Federico Del Pup","orcid":"https://orcid.org/0009-0004-0698-962X"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Del Pup, Federico","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"last","author":{"id":"https://openalex.org/A5075411914","display_name":"Manfredo Atzori","orcid":"https://orcid.org/0000-0001-5397-2063"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Atzori, Manfredo","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/T10036","display_name":"Advanced Neural Network Applications","score":0.8744000196456909,"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"}},"topics":[{"id":"https://openalex.org/T10036","display_name":"Advanced Neural Network Applications","score":0.8744000196456909,"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/T10052","display_name":"Medical Image Segmentation Techniques","score":0.012000000104308128,"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/T11307","display_name":"Domain Adaptation and Few-Shot Learning","score":0.010200000368058681,"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/computation","display_name":"Computation","score":0.5924000144004822},{"id":"https://openalex.org/keywords/segmentation","display_name":"Segmentation","score":0.5810999870300293},{"id":"https://openalex.org/keywords/transformer","display_name":"Transformer","score":0.5485000014305115},{"id":"https://openalex.org/keywords/encoder","display_name":"Encoder","score":0.5156000256538391},{"id":"https://openalex.org/keywords/computational-model","display_name":"Computational model","score":0.45899999141693115},{"id":"https://openalex.org/keywords/dice","display_name":"Dice","score":0.4016000032424927},{"id":"https://openalex.org/keywords/computational-complexity-theory","display_name":"Computational complexity theory","score":0.31839999556541443},{"id":"https://openalex.org/keywords/pattern-recognition","display_name":"Pattern recognition (psychology)","score":0.3174999952316284}],"concepts":[{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.7623999714851379},{"id":"https://openalex.org/C45374587","wikidata":"https://www.wikidata.org/wiki/Q12525525","display_name":"Computation","level":2,"score":0.5924000144004822},{"id":"https://openalex.org/C89600930","wikidata":"https://www.wikidata.org/wiki/Q1423946","display_name":"Segmentation","level":2,"score":0.5810999870300293},{"id":"https://openalex.org/C66322947","wikidata":"https://www.wikidata.org/wiki/Q11658","display_name":"Transformer","level":3,"score":0.5485000014305115},{"id":"https://openalex.org/C154945302","wikidata":"https://www.wikidata.org/wiki/Q11660","display_name":"Artificial intelligence","level":1,"score":0.5311999917030334},{"id":"https://openalex.org/C118505674","wikidata":"https://www.wikidata.org/wiki/Q42586063","display_name":"Encoder","level":2,"score":0.5156000256538391},{"id":"https://openalex.org/C66024118","wikidata":"https://www.wikidata.org/wiki/Q1122506","display_name":"Computational model","level":2,"score":0.45899999141693115},{"id":"https://openalex.org/C22029948","wikidata":"https://www.wikidata.org/wiki/Q45089","display_name":"Dice","level":2,"score":0.4016000032424927},{"id":"https://openalex.org/C119857082","wikidata":"https://www.wikidata.org/wiki/Q2539","display_name":"Machine learning","level":1,"score":0.33959999680519104},{"id":"https://openalex.org/C179799912","wikidata":"https://www.wikidata.org/wiki/Q205084","display_name":"Computational complexity theory","level":2,"score":0.31839999556541443},{"id":"https://openalex.org/C153180895","wikidata":"https://www.wikidata.org/wiki/Q7148389","display_name":"Pattern recognition (psychology)","level":2,"score":0.3174999952316284},{"id":"https://openalex.org/C195956108","wikidata":"https://www.wikidata.org/wiki/Q7268362","display_name":"Quadratic growth","level":2,"score":0.3138999938964844},{"id":"https://openalex.org/C98045186","wikidata":"https://www.wikidata.org/wiki/Q205663","display_name":"Process (computing)","level":2,"score":0.3102000057697296},{"id":"https://openalex.org/C113775141","wikidata":"https://www.wikidata.org/wiki/Q428691","display_name":"Computer engineering","level":1,"score":0.30809998512268066},{"id":"https://openalex.org/C73000952","wikidata":"https://www.wikidata.org/wiki/Q17007827","display_name":"Discretization","level":2,"score":0.3028999865055084},{"id":"https://openalex.org/C45347329","wikidata":"https://www.wikidata.org/wiki/Q5166604","display_name":"Convolution (computer science)","level":3,"score":0.299699991941452},{"id":"https://openalex.org/C79581498","wikidata":"https://www.wikidata.org/wiki/Q1367530","display_name":"Suite","level":2,"score":0.29760000109672546},{"id":"https://openalex.org/C123657996","wikidata":"https://www.wikidata.org/wiki/Q12271","display_name":"Architecture","level":2,"score":0.29170000553131104},{"id":"https://openalex.org/C2781357197","wikidata":"https://www.wikidata.org/wiki/Q5757597","display_name":"High memory","level":2,"score":0.2800000011920929},{"id":"https://openalex.org/C99821215","wikidata":"https://www.wikidata.org/wiki/Q1136583","display_name":"Swap (finance)","level":2,"score":0.2678999900817871},{"id":"https://openalex.org/C105339364","wikidata":"https://www.wikidata.org/wiki/Q2297740","display_name":"Software deployment","level":2,"score":0.26660001277923584},{"id":"https://openalex.org/C124504099","wikidata":"https://www.wikidata.org/wiki/Q56933","display_name":"Image segmentation","level":3,"score":0.25600001215934753}],"mesh":[],"locations_count":1,"locations":[{"id":"doi:10.48550/arxiv.2602.20008","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2602.20008","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":"Preprint"}],"best_oa_location":{"id":"doi:10.48550/arxiv.2602.20008","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2602.20008","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":false,"raw_source_name":null,"raw_type":"Preprint"},"sustainable_development_goals":[{"id":"https://metadata.un.org/sdg/9","score":0.4524865746498108,"display_name":"Industry, innovation and infrastructure"}],"awards":[],"funders":[],"has_content":{"pdf":false,"grobid_xml":false},"content_urls":null,"referenced_works_count":0,"referenced_works":[],"related_works":[],"abstract_inverted_index":{"We":[0],"present":[1],"Token-UNet,":[2],"adopting":[3],"the":[4,39,57,66,74,84,115,177,199,230,235,238],"TokenLearner":[5,123],"and":[6,88,105,121,138,162,174,220,228],"TokenFuser":[7],"modules":[8],"to":[9,73,124,171,201],"encase":[10],"Transformers":[11,15],"into":[12],"UNets.":[13],"While":[14],"have":[16],"enabled":[17],"global":[18,139],"interactions":[19],"among":[20],"input":[21,79],"elements":[22],"in":[23,101,205,222],"medical":[24,214],"imaging,":[25],"current":[26],"computational":[27,103,110,209],"challenges":[28],"hinder":[29],"their":[30],"deployment":[31],"on":[32],"common":[33],"hardware.":[34],"Models":[35],"like":[36],"(Swin)UNETR":[37],"adapt":[38],"UNet":[40],"architecture":[41],"by":[42],"incorporating":[43],"(Swin)Transformer":[44],"encoders,":[45],"which":[46,70],"process":[47],"tokens":[48,135],"that":[49,98],"each":[50],"represent":[51],"small":[52],"subvolumes":[53],"($8^3$":[54],"voxels)":[55],"of":[56,68,77,86,94,118,134,165,176,232,237],"input.":[58],"The":[59,155],"Transformer":[60],"attention":[61,153],"mechanism":[62],"scales":[63],"quadratically":[64],"with":[65,180,207],"number":[67,133],"tokens,":[69],"is":[71],"tied":[72],"cubic":[75],"scaling":[76],"3D":[78,95,125,213],"resolution.":[80],"This":[81,128,196],"work":[82,197],"reconsiders":[83],"role":[85],"convolution":[87],"attention,":[89],"introducing":[90],"Token-UNets,":[91],"a":[92,131],"family":[93],"segmentation":[96],"models":[97],"can":[99,226],"operate":[100],"constrained":[102],"environments":[104],"time":[106],"frames.":[107],"To":[108],"mitigate":[109],"demands,":[111],"our":[112,166,192],"approach":[113],"maintains":[114],"convolutional":[116],"encoder":[117],"UNet-like":[119],"models,":[120],"applies":[122],"feature":[126],"maps.":[127,154],"module":[129],"pools":[130],"preset":[132],"from":[136],"local":[137],"structures.":[140],"Our":[141],"results":[142],"show":[143],"this":[144],"tokenization":[145],"effectively":[146],"encodes":[147],"task-relevant":[148],"information,":[149],"yielding":[150],"naturally":[151],"interpretable":[152],"memory":[156],"footprint,":[157],"computation":[158],"times":[159],"at":[160],"inference,":[161],"parameter":[163],"counts":[164],"heaviest":[167],"model":[168,217],"are":[169],"reduced":[170],"33\\%,":[172],"10\\%,":[173],"35\\%":[175],"SwinUNETR":[178,190],"values,":[179],"better":[181],"average":[182],"performance":[183],"(86.75\\%":[184],"$\\pm":[185,194],"0.19\\%$":[186],"Dice":[187],"score":[188],"for":[189,234],"vs":[191],"87.21\\%":[193],"0.35\\%$).":[195],"opens":[198],"way":[200],"more":[202],"efficient":[203],"trainings":[204],"contexts":[206],"limited":[208,223],"resources,":[210],"such":[211],"as":[212],"imaging.":[215],"Easing":[216],"optimization,":[218],"fine-tuning,":[219],"transfer-learning":[221],"hardware":[224],"settings":[225],"accelerate":[227],"diversify":[229],"development":[231],"approaches,":[233],"benefit":[236],"research":[239],"community.":[240]},"counts_by_year":[],"updated_date":"2026-07-01T06:00:48.157686","created_date":"2026-02-26T00:00:00"}