arXivDaily arXiv每日学术速递 周一至周五更新

高校专区

ETH Zurich(苏黎世联邦理工学院)

2026-08-04 至 2026-08-04 共收录 2
2511.15503 2026-08-04 cs.AR cs.DC cs.LG cs.PF 版本更新

DCC: Data-Centric Compilation of Machine Learning Kernels for Processing-In-Memory Architectures

DCC: 面向处理-内存架构的机器学习内核数据驱动编译

Peiming Yang, Sankeerth Durvasula, Ivan Fernandez, Mohammad Sadrosadati, Onur Mutlu, Gennady Pekhimenko, Christina Giannoula

机构 * University of Toronto(多伦多大学) Vector Institute(向量研究所) Barcelona Supercomputing Center(巴塞罗那超级计算中心) ETH Zürich(苏黎世联邦理工学院) Nvidia(英伟达) Max Planck Institute for Software Systems(马克斯·普朗克软件系统研究所)

AI总结 针对PIM系统中数据重排与计算代码优化相互依赖的问题,提出数据驱动的编译器DCC,通过统一调优联合优化数据重排与计算代码,在HBM-PIM和AttAcc PIM上分别实现平均2.21倍和3.92倍的加速。

详情

展开后加载摘要…

URL PDF HTML 收藏
2510.09379 2026-08-04 cs.LG cs.AI cs.SY eess.SY 版本更新

Eigenvalues as a Metric for Memory Dynamics in Sequence Models

特征值作为序列模型记忆动力学的度量指标

Rahel Rickenbach, Jelena Trisovic, Alexandre Didier, Jerome Sieber, Melanie N. Zeilinger

机构 * ETH Zurich(苏黎世联邦理工学院) ETH AI Center(苏黎世人工智能中心)

AI总结 该研究提出以特征值为通用度量,通过动力系统框架统一分析注意力模型与状态空间模型的记忆动力学,经实证研究发现特征值可影响两类模型的记忆与长程依赖建模,还可用于指导架构修改与训练,为序列模型的分析与优化提供了新途径。

详情

展开后加载摘要…

URL PDF HTML 收藏