Saved in:
Bibliographic Details
Main Author: Kim, Sehun
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2410.14193
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866929590648176640
author Kim, Sehun
author_facet Kim, Sehun
contents A persistence diagram provides a compact summary of persistent homology, which captures the topological features of a space at different scales. However, due to its nature as a set, incorporating it as a feature into a machine learning framework is challenging. Several methods have been proposed to use persistence diagrams as input for machine learning models, but they often require complex preprocessing steps and extensive hyperparameter tuning. In this paper, we propose a novel transformer architecture called the \textit{Extended Persistence Transformer (xPerT)}, which is highly scalable than the compared to Persformer, an existing transformer for persistence diagrams. xPerT reduces GPU memory usage by over 90\% and improves accuracy on multiple datasets. Additionally, xPerT does not require complex preprocessing steps or extensive hyperparameter tuning, making it easy to use in practice. Our code is available at https://github.com/sehunfromdaegu/xpert.
format Preprint
id arxiv_https___arxiv_org_abs_2410_14193
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle xPerT: Extended Persistence Transformer
Kim, Sehun
Machine Learning
Algebraic Topology
A persistence diagram provides a compact summary of persistent homology, which captures the topological features of a space at different scales. However, due to its nature as a set, incorporating it as a feature into a machine learning framework is challenging. Several methods have been proposed to use persistence diagrams as input for machine learning models, but they often require complex preprocessing steps and extensive hyperparameter tuning. In this paper, we propose a novel transformer architecture called the \textit{Extended Persistence Transformer (xPerT)}, which is highly scalable than the compared to Persformer, an existing transformer for persistence diagrams. xPerT reduces GPU memory usage by over 90\% and improves accuracy on multiple datasets. Additionally, xPerT does not require complex preprocessing steps or extensive hyperparameter tuning, making it easy to use in practice. Our code is available at https://github.com/sehunfromdaegu/xpert.
title xPerT: Extended Persistence Transformer
topic Machine Learning
Algebraic Topology
url https://arxiv.org/abs/2410.14193