- 博客(20)
- 收藏
- 关注
原创 RKNN-Toolkit2 转换脚本
在脚本结束时调用是好习惯,但在 Jupyter Notebook 中也可以省略。工具查看模型结构、算子分布等信息,确认是否符合目标芯片。,可以把这两个参数留空,让 SDK 自动处理。路径可以是绝对路径,也可以相对于当前工作目录。:如果模型里有多路输入,记得在。:导出后可以用 SDK 自带的。释放 RKNN 对象占用的内存。:如果你在模型里已经做了。:若你不想做量化,只需在。
2025-09-06 22:31:54
239
原创 NPU:AI时代的计算核心与架构演进
从诞生到多代演进,NPU的发展之路清晰地展示了计算架构“术业有专攻”的理念。它并非CPU或GPU的替代品,而是一个不可或缺的协作者,共同构成了现代高性能异构计算平台。正是有了NPU,复杂的AI算法才能走出云端数据中心,在我们身边的智能设备中高效、低功耗地运行,真正赋能万物互联的智能时代。
2025-09-06 18:29:01
254
原创 RKNN 3588 RKNPU 的算力到底怎么算?
是专门为 AI 推理/训练设计的加速器。它在每个时钟周期里能做(MAC) 操作——一次乘法 + 一次加法。
2025-09-06 17:55:31
460
原创 【几何深度学习(Geometric Deep Learning)】几何深度学习与群论:从对称性到可解释的图网络关键词
本文探讨了群论在几何深度学习中的应用,重点分析了对称性、不可变性和共变性对图神经网络(GNN)设计的影响。文章回顾了GCN、GAT和GraphSAGE等核心算法如何实现这些群论原则,并阐述了将物理对称性嵌入模型带来的性能提升和可解释性增强。研究表明,基于群论的GNN设计能有效减少学习负担、提升泛化能力,为AI与科学计算交叉领域(如量子化学、材料设计)提供了新思路。未来工作可进一步探索连续群在GNN中的实现,深化物理约束与深度学习模型的融合。
2025-09-01 14:25:23
241
原创 【几何深度学习Geometric Deep Learning】跨模态几何融合
跨模态几何融合正成为多感知系统的核心技术之一。通过将图、点云与序列等非欧式数据统一映射到共享嵌入空间,并利用异构图 + Transformer 的联合框架,可以在保持各自几何不变性的前提下,实现强大的跨模态交互和推理能力。未来的研究应聚焦于更高效的投影方法、更鲁棒的对齐策略以及可解释性与实时部署之间的平衡,以推动 CGF 在自动驾驶、机器人感知、生物信息学等领域的深入应用。
2025-09-01 14:11:46
202
原创 【几何深度学习(Geometric Deep Learning)】大规模动态图(Dynamic Graphs)
给出一套完整的技术方案,既能保证图结构与属性的“对称性”——即无论从哪个方向读取信息都得到相同结果,又能在大规模动态图(数十亿级别顶点/边)中实现高效、可扩展的特征更新。:研究人员、算法工程师以及需要构建实时图服务的系统架构师。(含特征更新),且对称性误差低于 1e−61e−6。实验表明,在 AWS EMR 上使用上述方案,
2025-09-01 14:03:08
129
原创 【几何深度学习(Geometric Deep Learning)】高阶群的可微实现
在深度学习框架(PyTorch / TensorFlow)中构造可微、数值稳定且计算高效的分解与参数化方法,以便直接在梯度下降过程中使用这些群元素。
2025-09-01 13:52:12
101
原创 【几何深度学习(Geometric Deep Learning)】统一框架:GNN 与 GDL 的关系
统一框架:传统 DL 的卷积/循环/自注意力在 GDL 中被映射为图卷积、交互式注意力和局部 Transformer,核心归纳偏置从平移变为图的自同构。理论深度:GCN 基于谱域拉普拉斯正则化;GAT 通过可学习的注意力实现节点间加权聚合;Graph Transformer 在保持全局信息的同时,利用窗口机制控制稀疏性。对称性与可解释性:对称群是 GDL 的核心归纳偏置;网络结构直接对应物理/几何约束,使模型更易于解释和调优。未来方向多模态图学习:融合图、时间序列、文本等信息;可微分对称性学习。
2025-09-01 13:34:02
233
原创 【几何深度学习(Geometric Deep Learning)】物理对称性嵌入网络
一个的完整示例,演示如何把物理对称性(这里是 3‑D 空间中的 SO(3) 旋转)嵌入到深度网络里。我们使用的是e3nn(一个专门为 SE(3)-equivariant 网络提供的 PyTorch 生态),并结合来处理分子图数据。
2025-09-01 13:19:43
171
原创 【几何深度学习(Geometric Deep Learning)】MeshCNN, Graph U-Net
本文提供了两个基于图神经网络的完整PyTorch实现示例: MeshCNN示例:在三角网格边上进行1D卷积操作,包含边卷积层(EdgeConv)、网格池化层(MeshPool)和网格上采样层(MeshUnpool)。关键点包括使用index_select高效处理边索引映射,预计算池化映射提升性能,并采用[B,C,E]的张量维度布局与Conv1d兼容。 Graph-U-Net示例:实现包含图卷积(GCNConv)和TopK池化的U-Net结构,通过存储池化索引实现上采样对齐,并利用跳跃连接保持特征维度一致。
2025-09-01 12:23:20
176
原创 【几何深度学习 Geometric Deep Learning】PointNet & PointNet++ (点云)
本文提供了一个完整的PyTorch实现,包含PointNet和PointNet++两种点云处理网络。代码实现了数据读取(ModelNet40数据集)、网络结构(Farthest Point Sampling采样、Ball Query分组、PointNet层等核心模块)以及训练/验证流程。其中PointNet++采用分层局部聚合策略,通过Set Abstraction和Feature Propagation层实现特征提取。代码在PyTorch1.13+CUDA11.x环境下测试通过,支持GPU/CPU切换,包
2025-09-01 12:17:08
105
原创 【几何深度学习 Geometric Deep Learning】Graph Transformer
本文介绍了GraphTransformer的核心思想与实现方法。其关键创新在于:1)采用邻域注意力机制,仅在节点局部子图中进行交互,避免传统自注意力的O(N²)计算开销;2)引入位置编码(如最短路径距离或随机游走)来保留图拓扑结构。文章提供了完整的PyTorch实现,包括数据准备、邻域注意力层、位置编码模块和完整的GraphTransformer模型架构,并给出了训练循环示例。该实现支持多头注意力、残差连接和层归一化等现代Transformer设计,可直接应用于节点分类等任务,同时讨论了梯度消失、内存占用等
2025-09-01 12:04:37
152
原创 【几何深度学习 Geometric Deep Learning】GraphSAGE 代码示例
本文介绍了GraphSAGE在大规模图上的实现方法,重点演示了通过邻居采样降低计算复杂度的PyTorch代码实现。文章使用PyTorch Geometric库,详细展示了数据准备、模型构建、邻居采样器和训练循环的实现过程,包含完整的代码示例和关键参数说明。同时提供了纯PyTorch实现思路,并推荐了常见优化技巧,如梯度累积、混合精度等。最后强调邻居采样是GraphSAGE可扩展性的核心,PyG的NeighborLoader封装了采样、子图构造和批处理,只需几行代码即可实现高效训练。
2025-09-01 11:56:27
103
原创 【几何深度学习(Geometric Deep Learning】Graph Attention Network(GAT)完整实现
本文介绍了PyTorch版GraphAttentionNetwork(GAT)的完整实现,包含数据准备、模型构建、训练优化等关键环节。使用Planetoid的Cora数据集,实现了一个包含多头注意力机制的两层GAT模型,采用Dropout、残差连接、学习率调度等优化技巧。训练过程记录了损失和准确率,并可视化学习曲线。文章还提供了训练结果示例(Cora数据集上达到约90%准确率)以及针对内存占用、梯度问题等常见挑战的优化建议。最后推荐了相关论文和扩展阅读方向。
2025-09-01 11:48:58
179
原创 【几何深度学习(Geometric Deep Learning)】Graph Convolutional Network(GCN)实现示例
一个的 Graph Convolutional Network(GCN)实现示例,使用torch。代码从数据加载 → 模型定义 → 训练循环 → 验证/测试,到常用优化技巧与可视化等步骤都做了详细注释,方便你直接复制粘贴后跑起来,也可以作为参考模板在自己的项目里快速改造。:节点分类(Cora 示例):2‑层 GCN + Dropout + BatchNorm:学习率调度、权重衰减、EarlyStopping、混合精度训练、GPU 多卡等。
2025-09-01 11:41:40
129
原创 几何深度学习(Geometric Deep Learning)—— 原理与核心算法
主要内容包括:几何深度学习的概念框架、对称性与不变性的数学基础、卷积、注意力及图网络中的群理论实现,以及常见模型(GCN、GAT、SAGE、Graph Transformer、PointNet++ 等)的核心原理。全文以**“从先验到归纳”**的视角展开,帮助读者把握如何将物理对称性或几何约束嵌入网络结构。
2025-09-01 11:34:08
642
原创 逆卷积(Reverse Convolution)与图像恢复
下面展示如何用 PyTorch 实现 RC 层,并把它嵌入到一个简易的 U‑Net 模型里。:相比传统的反卷积(deconv)或上采样+卷积,RC 直接从深层特征得到像素级别信息,无需额外的插值操作,减少了模糊与伪影。感兴趣,可以直接查看官方 GitHub 代码并在自己的项目中实验,或者尝试把它与其他网络结构结合,探索更广阔的应用场景。—— 这是一种“反向卷积”,用来把深度特征恢复到像素级别。如果只用正向卷积,很难直接从高维特征“逆推”到低维的重建结果。的:先对特征进行卷积,再得到更高层次的表征。
2025-09-01 00:13:48
186
原创 BED(Bayesian Evidence Decoding) 任务的新型网络设计
一个,演示如何把⚠️ 为了保持代码简洁,所有的实验设置(如数据、损失等)都做了最小化处理;你可以直接把这段代码粘到并运行。若想获得更高性能,请根据自己的需求调整超参数或改进网络结构。
2025-09-01 00:06:20
78
一个高性能的 LNN-CfC 模型
2026-01-30
基于物理信息的 Deep-ESN 用于频率捷变雷达微弱目标检测
2026-01-24
图神经网络示例代码-第一至6章
2025-11-19
AdaI-PINNs:高效求解界面问题的自适应物理信息神经网络框架
2025-11-18
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人
RSS订阅