A User's Guide to $\texttt{KSig}$: GPU-Accelerated Computation of the Signature Kernel

Csaba Tóth, Danilo Jr Dela Cruz, Harald Oberhauser
1/13/2025
stat.MLcs.LG

Official code: tgcsaba/KSig (52 stars, Apache-2.0, last commit 2025-06-06).

Abstract

The signature kernel is a positive definite kernel for sequential and temporal data that has become increasingly popular in machine learning applications due to powerful theoretical guarantees, strong empirical performance, and recently introduced various scalable variations. In this chapter, we give a short introduction to $\texttt{KSig}$, a $\texttt{Scikit-Learn}$ compatible Python package that implements various GPU-accelerated algorithms for computing signature kernels, and performing downstream learning tasks. We also introduce a new algorithm based on tensor sketches which gives strong performance compared to existing algorithms. The package is available at https://github.com/tgcsaba/ksig.

View on arXivView PDF

Code Implementations(1)

tgcsaba/KSigOfficial100%
Apache-2.0

A scikit-learn compatible Python package for GPU-accelerated computation of the signature kernel using CuPy.

527Python, Jupyter NotebookMay 10, 20211 years agoApache-2.0

Similar Papers

Cite this paper

@article{tth2025users,
  title  = {A User's Guide to $\texttt{KSig}$: GPU-Accelerated Computation of the Signature Kernel},
  author = {Csaba Tóth and Danilo Jr Dela Cruz and Harald Oberhauser},
  year   = {2025},
  eprint = {2501.07145},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2501.07145v2}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2501.07145.svg)](https://codeofpaper.com/paper/2501.07145)

Discussion