English
Related papers

Related papers: Interpretable Lightweight Transformer via Unrollin…

200 papers

Unlike conventional "black-box" transformers with classical self-attention mechanism, we build a lightweight and interpretable transformer-like neural net by unrolling a mixed-graph-based optimization algorithm to forecast traffic with…

Machine Learning · Computer Science 2026-02-04 Ji Qi , Tam Thuc Do , Mingxiao Liu , Zhuoshi Pan , Yuzhe Li , Gene Cheung , H. Vicky Zhao

Generic deep learning (DL) networks for image restoration like denoising and interpolation lack mathematical interpretability, require voluminous training data to tune a large parameter set, and are fragile in the face of covariate shift.…

Image and Video Processing · Electrical Eng. & Systems 2025-03-13 Jianghe Cai , Gene Cheung , Fei Chen

We propose an interpretable graph neural network framework to denoise single or multiple noisy graph signals. The proposed graph unrolling networks expand algorithm unrolling to the graph domain and provide an interpretation of the…

Signal Processing · Electrical Eng. & Systems 2021-09-08 Siheng Chen , Yonina C. Eldar , Lingxiao Zhao

An image denoiser can be used for a wide range of restoration problems via the Plug-and-Play (PnP) architecture. In this paper, we propose a general framework to build an interpretable graph-based deep denoiser (GDD) by unrolling a solution…

Image and Video Processing · Electrical Eng. & Systems 2024-09-11 Seyed Alireza Hosseini , Tam Thuc Do , Gene Cheung , Yuichi Tanaka

Given partial measurements of a time-varying graph signal, we propose an algorithm to simultaneously estimate both the underlying graph topology and the missing measurements. The proposed algorithm operates by training an interpretable…

Signal Processing · Electrical Eng. & Systems 2024-07-17 Subbareddy Batreddy , Pushkal Mishra , Yaswanth Kakarla , Aditya Siripuram

While neural networks are powerful approximators used to classify or embed data into lower dimensional spaces, they are often regarded as black boxes with uninterpretable features. Here we propose Graph Spectral Regularization for making…

We propose a probabilistic interpretation of transformers as unrolled inference steps assuming a probabilistic Laplacian Eigenmaps model from the ProbDR framework. Our derivation shows that at initialisation, transformers perform "linear"…

Machine Learning · Computer Science 2025-07-29 Aditya Ravuri , Neil D. Lawrence

Despite the popularity of transformers in practice, their architectures are empirically designed and neither mathematically justified nor interpretable. Moreover, as indicated by many empirical studies, some components of transformer…

Machine Learning · Computer Science 2025-06-05 Peng Wang , Yifu Lu , Yaodong Yu , Druv Pai , Qing Qu , Yi Ma

In this paper, we explore the topic of graph learning from the perspective of the Irregularity-Aware Graph Fourier Transform, with the goal of learning the graph signal space inner product to better model data. We propose a novel method to…

Signal Processing · Electrical Eng. & Systems 2023-03-16 Benjamin Girault , Eduardo Pavez , Antonio Ortega

We propose a generalization of transformer neural network architecture for arbitrary graphs. The original transformer was designed for Natural Language Processing (NLP), which operates on fully connected graphs representing all connections…

Machine Learning · Computer Science 2021-01-26 Vijay Prakash Dwivedi , Xavier Bresson

Graph Neural Networks (GNNs) have become a standard approach for learning from graph-structured data. However, their reliance on parametric classifiers (most often linear softmax layers) limits interpretability and sometimes hinders…

Machine Learning · Computer Science 2026-02-03 Zeljko Bolevic , Milos Brajovic , Isidora Stankovic , Ljubisa Stankovic

Graph Neural Networks (GNNs) are the dominant architecture for molecular machine learning, particularly for molecular property prediction and machine learning interatomic potentials (MLIPs). GNNs perform message passing on predefined graphs…

Machine Learning · Computer Science 2025-10-03 Tobias Kreiman , Yutong Bai , Fadi Atieh , Elizabeth Weaver , Eric Qu , Aditi S. Krishnapriyan

Graphs serve as generic tools to encode the underlying relational structure of data. Often this graph is not given, and so the task of inferring it from nodal observations becomes important. Traditional approaches formulate a convex inverse…

Machine Learning · Computer Science 2024-06-24 Max Wasserman , Gonzalo Mateos

Graph neural networks (GNNs) achieve strong performance on graph learning tasks, but training on large-scale networks remains computationally challenging. Transferability results show that GNNs with fixed weights can generalize from smaller…

Signal Processing · Electrical Eng. & Systems 2026-04-17 Haoyu Wang , Renyuan Ma , Gonzalo Mateos , Luana Ruiz

Data-driven surrogate modeling has surged in capability in recent years with the emergence of graph neural networks (GNNs), which can operate directly on mesh-based representations of data. The goal of this work is to introduce an…

Machine Learning · Computer Science 2024-10-25 Shivam Barwey , Hojin Kim , Romit Maulik

Low-rank approximation models of data matrices have become important machine learning and data mining tools in many fields including computer vision, text mining, bioinformatics and many others. They allow for embedding high-dimensional…

Machine Learning · Computer Science 2020-10-19 Penglong Zhai , Shihua Zhang

In this work, we provide a new formulation for Graph Convolutional Neural Networks (GCNNs) for link prediction on graph data that addresses common challenges for biomedical knowledge graphs (KGs). We introduce a regularized attention…

Machine Learning · Computer Science 2018-12-04 Daniel Neil , Joss Briody , Alix Lacoste , Aaron Sim , Paidi Creed , Amir Saffari

We introduce a method for fast estimation of data-adapted, spatio-temporally dependent regularization parameter-maps for variational image reconstruction, focusing on total variation (TV)-minimization. Our approach is inspired by recent…

Standard deep learning architectures used for classification generate label predictions with a projection head and softmax activation function. Although successful, these methods fail to leverage the relational information between samples…

Machine Learning · Computer Science 2025-12-10 Jason Brown , Bohan Chen , Harris Hardiman-Mostow , Jeff Calder , Andrea L. Bertozzi

Conventional deep neural nets (DNNs) initialize network parameters at random and then optimize each one via stochastic gradient descent (SGD), resulting in substantial risk of poor-performing local minima. Focusing on image interpolation…

Computer Vision and Pattern Recognition · Computer Science 2026-04-08 Xue Zhang , Bingshuo Hu , Gene Cheung
‹ Prev 1 2 3 10 Next ›