计算机科学 > 计算机视觉与模式识别
[提交于 2025年8月29日
]
标题: GraViT:使用视觉变压器和MLP-Mixer进行强引力透镜发现的迁移学习
标题: GraViT: Transfer Learning with Vision Transformers and MLP-Mixer for Strong Gravitational Lens Discovery
摘要: 引力透镜效应为研究暗物质的特性提供了一个强大的探针,并对于推断宇宙学参数至关重要。 空间和时间的遗产调查(LSST)预计在未来十年内发现O(10^5)个引力透镜,这需要自动分类器。 在本工作中,我们介绍了GraViT,这是一个用于引力透镜检测的PyTorch管道,它利用了最先进的视觉变换器(ViT)模型和MLP-Mixer的广泛预训练。 我们通过检查数据质量(源和样本量)、模型架构(选择和微调)、训练策略(增强、归一化和优化)和集成预测来评估迁移学习对分类性能的影响。 本研究重现了之前神经网络系统比较中的实验,并提供了关于在该通用测试样本上强引力透镜可检测性的见解。 我们使用来自HOLISMOKES VI和SuGOHI X的数据集对十个架构进行微调,并将其与卷积基线进行基准测试,讨论复杂性和推理时间分析。
文献和引用工具
与本文相关的代码,数据和媒体
alphaXiv (什么是 alphaXiv?)
CatalyzeX 代码查找器 (什么是 CatalyzeX?)
DagsHub (什么是 DagsHub?)
Gotit.pub (什么是 GotitPub?)
Hugging Face (什么是 Huggingface?)
带有代码的论文 (什么是带有代码的论文?)
ScienceCast (什么是 ScienceCast?)
演示
推荐器和搜索工具
arXivLabs:与社区合作伙伴的实验项目
arXivLabs 是一个框架,允许合作伙伴直接在我们的网站上开发和分享新的 arXiv 特性。
与 arXivLabs 合作的个人和组织都接受了我们的价值观,即开放、社区、卓越和用户数据隐私。arXiv 承诺这些价值观,并且只与遵守这些价值观的合作伙伴合作。
有一个为 arXiv 社区增加价值的项目想法吗? 了解更多关于 arXivLabs 的信息.