单张 GB200 聚类 10 万金融工具,64 卡扩展至百万级
PyTorch · x · 2026-09-11
PyTorch 官方转引 NVIDIA 技术博客,介绍 GPU 加速的金融工具聚类管线 AdaptGrow:
- 方法:基于对称非负矩阵分解(SymNMF),从滚动相关矩阵和尾部成对依赖矩阵(TPDM)中产出硬聚类标签、软因子载荷与结构突变信号
- 显存优化:峰值存储从约 20n² 降至 4n² 字节,单张 GB200 可对约 10 万个工具做分解
- 分布式扩展:按行分片依赖矩阵、经 NCCL 通信 O(nk) 数据,64 张 GB200(16 节点)可扩展到 100 万个工具
- 自动配置:根据矩阵特征谱自动选择 full-batch AdaGrad(谱间隙清晰)或 block-stochastic SVRG(谱较平),单卡到多卡接口一致
- 稳定性追踪:250 个滚动窗口(约一个交易年)中用自校准 3σ 控制限的调整兰德指数(ARI)检测成员变动;TPDM 还能捕捉相关性遗漏的跨行业共跌事件
附完整可复现 notebook(clusteringthroughtime.ipynb)与 SymNMF-factors 仓库。
「Infra」频道最新
- Anthropic S-1:每 1 美元收入烧 48 美分算力成本 — nmasc_ · 2026-10-26
- Persistent AI 入驻 UR 网络,隐私推理流量全程加密 — 0xSammy · 2026-10-26
- Anthropic 算力承诺曝光:Google 至少 1145 亿美元居首 — BenBajarin · 2026-10-26
- SpaceX 与 Apollo 达成 400 亿美元 AI 芯片与数据中心硬件租赁协议 — Polymarket · 2026-10-26
- Diamandis 盘点:SpaceX AI 业务年增 830%,Claude 64 小时设计 31.8 万抗毒蛋白 — PeterDiamandis · 2026-10-26
- Meta 拟发售至多 950 亿美元股票为 AI 基建输血 — The Decoder · 2026-10-26