Saved in:
Bibliographic Details
Main Authors: Wang, Elena Xinyi, Nigmetov, Arnur, Morozov, Dmitriy
Format: Preprint
Published: 2026
Subjects:
Online Access:https://arxiv.org/abs/2604.08469
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866917419867439104
author Wang, Elena Xinyi
Nigmetov, Arnur
Morozov, Dmitriy
author_facet Wang, Elena Xinyi
Nigmetov, Arnur
Morozov, Dmitriy
contents Topological Data Analysis (TDA) provides tools to describe the shape of data, but integrating topological features into deep learning pipelines remains challenging, especially when preserving local geometric structure rather than summarizing it globally. We propose a persistence-based data augmentation framework that encodes local gradient flow regions and their hierarchical evolution using the Morse-Smale complex. This representation, compatible with both convolutional and graph neural networks, retains spatially localized topological information across multiple scales. Importantly, the augmentation procedure itself is efficient, with computational complexity $O(n \log n)$, making it practical for large datasets. We evaluate our method on histopathology image classification and 3D porous material regression, where it consistently outperforms baselines and global TDA descriptors such as persistence images and landscapes. We also show that pruning the base level of the hierarchy reduces memory usage while maintaining competitive performance. These results highlight the potential of local, structured topological augmentation for scalable and interpretable learning across data modalities.
format Preprint
id arxiv_https___arxiv_org_abs_2604_08469
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Persistence-Augmented Neural Networks
Wang, Elena Xinyi
Nigmetov, Arnur
Morozov, Dmitriy
Machine Learning
Topological Data Analysis (TDA) provides tools to describe the shape of data, but integrating topological features into deep learning pipelines remains challenging, especially when preserving local geometric structure rather than summarizing it globally. We propose a persistence-based data augmentation framework that encodes local gradient flow regions and their hierarchical evolution using the Morse-Smale complex. This representation, compatible with both convolutional and graph neural networks, retains spatially localized topological information across multiple scales. Importantly, the augmentation procedure itself is efficient, with computational complexity $O(n \log n)$, making it practical for large datasets. We evaluate our method on histopathology image classification and 3D porous material regression, where it consistently outperforms baselines and global TDA descriptors such as persistence images and landscapes. We also show that pruning the base level of the hierarchy reduces memory usage while maintaining competitive performance. These results highlight the potential of local, structured topological augmentation for scalable and interpretable learning across data modalities.
title Persistence-Augmented Neural Networks
topic Machine Learning
url https://arxiv.org/abs/2604.08469