arXiv:2510.14516cs.CV2025-10被引 1

用Vision Mamba预测三维多孔介质渗透率,更高效省内存。

Vision Mamba for Permeability Prediction of Porous Media

  • 以Vision Mamba为骨干网络,替代ViT和CNN进行渗透率预测
  • 相比ViT和CNN,计算量更低、参数更少,精度相当或更高
  • 适合需要高效视觉建模的地质/材料仿真研究者

Vision Mamba作为图像分类的新兴架构,相较视觉变换器(ViTs)具有线性扩展的网络规模,而后者呈二次增长,显著提升计算与内存效率。同时,Vision Mamba所需可训练参数远少于传统卷积神经网络(CNNs),具备更强的内存效率。本文首次将Vision Mamba引入三维多孔介质渗透率预测任务,对比其在多个维度上的表现,评估其组件对精度的影响。实验表明,Vision Mamba在渗透率预测中兼具高效性与高精度,优于现有方法。源代码已公开,支持复现与后续研究。该框架有望嵌入大规模视觉模型,以Vision Mamba替代ViTs。

原文摘要 · Abstract (English)

Vision Mamba has recently received attention as an alternative to Vision Transformers (ViTs) for image classification. The network size of Vision Mamba scales linearly with input image resolution, whereas ViTs scale quadratically, a feature that improves computational and memory efficiency. Moreover, Vision Mamba requires a significantly smaller number of trainable parameters than traditional convolutional neural networks (CNNs), and thus, they can be more memory efficient. Because of these features, we introduce, for the first time, a neural network that uses Vision Mamba as its backbone for predicting the permeability of three-dimensional porous media. We compare the performance of Vision Mamba with ViT and CNN models across multiple aspects of permeability prediction and perform an ablation study to assess the effects of its components on accuracy. We demonstrate in practice the aforementioned advantages of Vision Mamba over ViTs and CNNs in the permeability prediction of three-dimensional porous media. We make the source code publicly available to facilitate reproducibility and to enable other researchers to build on and extend this work. We believe the proposed framework has the potential to be integrated into large vision models in which Vision Mamba is used instead of ViTs.

多孔介质渗透率预测Vision Mamba高效建模

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。