← Search

Ziheng Chen

23 accepted papers

2026

Fast and Stable Riemannian Metrics on SPD Manifolds via Cholesky Product Geometry

ICLR 2026poster

Recent advances in Symmetric Positive Definite (SPD) matrix learning show that Riemannian metrics are fundamental to effective SPD neural networks. Motivated by this, we revisit the geometry of the Cholesky factors and uncover a simple product structure that enables convenient metric design. Buildin…

Cited by 0SourcecodeScholar
2026

MeSH: Memory-as-State-Highways for Recursive Transformers

ICLR 2026poster

Recursive transformers reuse parameters and iterate over hidden states multiple times, decoupling compute depth from parameter depth. However, under matched compute, recursive models with fewer parameters often lag behind non-recursive counterparts. By probing hidden states, we trace this performanc…

Cited by 0SourcecodeScholar
2026

Riemannian Graph Convolutional Network for Skeleton-Based Two-Person Interaction Recognition

IJCAI 2026

In the field of skeleton-based human action recognition, Graph Convolutional Networks (GCNs) have become a dominant framework. However, existing GCN-based approaches often treat the sequences of two-person interaction as separate entities, ignoring the inherent semantic dependencies and spatial corr

Cited by 0Scholar
2026

Riemannian High-Order Pooling for Brain Foundation Models

ICLR 2026poster

Electroencephalography (EEG) is a noninvasive technique for measuring brain electrical activity that supports a wide range of brain-computer interaction applications. Motivated by the breakthroughs of Large Language Models (LLMs), recent efforts have begun to explore Large EEG foundation Models trai…

Cited by 0SourcecodeScholar
2026

Wasserstein-Aligned Hyperbolic Multi-View Clustering

AAAI 2026technical

Multi-view clustering (MVC) aims to uncover the latent structure of multi-view data by learning view-common and view-specific information. Although recent studies have explored hyperbolic representations for better tackling the representation gap between different views, they focus primarily on inst

Cited by 0SourcePDFScholar
2025

A Correlation Manifold Self-Attention Network for EEG Decoding

IJCAI 2025

Riemannian neural networks, which generalize the deep learning paradigm to non-Euclidean geometries, have garnered widespread attention across diverse applications in artificial intelligence. Among these, the representative attention models have been studied on various non-Euclidean spaces to geomet

2025

Learning to Normalize on the SPD Manifold under Bures-Wasserstein Geometry

CVPR 2025poster

Covariance matrices have proven highly effective across many scientific fields. Since these matrices lie within the Symmetric Positive Definite (SPD) manifold--a Riemannian space with intrinsic non-Euclidean geometry, the primary challenge in representation learning is to respect this underlying geo…

2025

Towards a General Attention Framework on Gyrovector Spaces for Matrix Manifolds

NeurIPS 2025poster

Deep neural networks operating on non-Euclidean geometries have recently demonstrated impressive performance across various machine-learning applications. Several studies have extended the attention mechanism to different manifolds. However, most existing non-Euclidean attention models are tailored…

Cited by 0SourceScholar
2025

Understanding Matrix Function Normalizations in Covariance Pooling through the Lens of Riemannian Geometry

ICLR 2025poster

Global Covariance Pooling (GCP) has been demonstrated to improve the performance of Deep Neural Networks (DNNs) by exploiting second-order statistics of high-level representations. GCP typically performs classification of the covariance matrices by applying matrix function normalization, such as mat…

2024

A Grassmannian Manifold Self-Attention Network for Signal Classification

IJCAI 2024poster

In the community of artificial intelligence, significant progress has been made in encoding sequential data using deep learning techniques. Nevertheless, how to effectively mine useful information from channel dimensions remains a major challenge, as these features have a submanifold structure. Line…

2024

RMLR: Extending Multinomial Logistic Regression into General Geometries

NeurIPS 2024poster

Riemannian neural networks, which extend deep learning techniques to Riemannian spaces, have gained significant attention in machine learning. To better classify the manifold-valued features, researchers have started extending Euclidean multinomial logistic regression (MLR) into Riemannian manifolds…

2024

Research on Autonomous Navigation of Dual-mode Wheel-legged Robot

IROS 2024poster

In order to improve the terrain adaptability and energy efficiency of wheel-legged robot in complex environment, a dual-mode navigation system based on robot energy consumption model is proposed. Firstly, the obstacle trafficability is evaluated according to the maximum obstacle crossing capability…

Cited by 0SourceScholar
2024

Riemannian Multinomial Logistics Regression for SPD Neural Networks

CVPR 2024poster

Deep neural networks for learning Symmetric Positive Definite (SPD) matrices are gaining increasing attention in machine learning. Despite the significant progress most existing SPD networks use traditional Euclidean classifiers on an approximated space rather than intrinsic classifiers that accurat…

2023

Riemannian Local Mechanism for SPD Neural Networks

AAAI 2023technical

The Symmetric Positive Definite (SPD) matrices have received wide attention for data representation in many scientific areas. Although there are many different attempts to develop effective deep architectures for data processing on the Riemannian manifold of SPD matrices, very few solutions explicit…

2022

AdaLoGN: Adaptive Logic Graph Network for Reasoning-Based Machine Reading Comprehension

ACL 2022long

Recent machine reading comprehension datasets such as ReClor and LogiQA require performing logical reasoning over text. Conventional neural models are insufficient for logical reasoning, while symbolic reasoners cannot directly apply to text. To meet the challenge, we present a neural-symbolic appro…