稠密矩阵和稀疏矩阵的对比

稠密矩阵和稀疏矩阵的对比

计算图的邻接矩阵时跑不动,爆内存,运行自动停止:

python 复制代码
GPU state:  True
Dataset: assist2012, Ques number: 53091, Skill number: 265

Killed

Dense(稠密矩阵)

稠密矩阵(dense matrix)就是每个元素都显式存储在内存里的矩阵。

例如一个 3×3 的矩阵:

A = \[1, 0, 2,

0, 0, 3,

4, 0, 5\]

在稠密矩阵里,即使很多元素是 0,0 也会占内存存储。

特点:

内存占用 = 元素个数 × 每个元素大小

矩阵乘法、加法等操作都是针对所有元素进行的

小矩阵无所谓,但大矩阵(尤其是几百万 × 几百万)会 爆内存

Sparse(稀疏矩阵)

**稀疏矩阵(sparse matrix)**只存储 非零元素及其索引

上面的矩阵,如果只存非零元素,会存成:

值: 1,2,3,4,5

行索引: 0,0,1,2,2

列索引: 0,2,2,0,2

内存占用 ≈ 非零元素数 × (值+索引)

当矩阵大而且大部分是 0 时,稀疏矩阵内存节省巨大

3️⃣ 为什么你的 G 计算爆内存

你原来的代码:

G = DV2 * H * W * invDE * HT * DV2

H 是 (百万 × 几百)

HT 是 (几百 × 百万)

当你相乘时,中间结果就成了 (百万 × 百万) 的矩阵 dense

即便大部分元素是 0,也会被完整存储,导致内存瞬间 TB 级别 → 爆掉

4️⃣ 总结

Dense 计算 = 所有元素都存储和计算,即使是 0

Sparse 计算 = 只存储非零元素,用稀疏算法计算,节省内存

对你的超图 G,必须全程稀疏矩阵计算,否则不可能在普通电脑上跑

相关推荐
zx_7414848113 分钟前
【机器学习入门】PyTorch 神经网络、卷积神经网络 CNN 实现矿物分类
pytorch·神经网络·机器学习
今天AI了吗14 分钟前
Codex 配置自定义 AI API 完整指南:从零到一接入你的专属模型
java·人工智能·python·数据分析·embedding
阿童木写作20 分钟前
跨境电商批量图片翻译与视频字幕翻译工具推荐
python·音视频
leihefeng28 分钟前
手写数字识别:KNN vs 逻辑回归实战
python·算法·机器学习·逻辑回归·scikit-learn
VL——MOESR30 分钟前
【具身智能】TurboVLA阅读随笔
论文阅读·机器学习·turbo·具身智能·vla
SunnyDays101140 分钟前
如何使用 Python 从 Word 文档中提取图片
python·word
2601_962097361 小时前
Pathway 实时RAG新方案:抛弃Spark、Flink,生产级数据管道
python·流处理·大数据集群·实时rag·pathway
张欣-男1 小时前
5分钟理解线性代数v2
人工智能·线性代数·机器学习
三十岁老牛再出发1 小时前
9月3日总结
python
2601_962299881 小时前
Linux下运行Python脚本
linux·python·ubuntu·脚本·命令