English
Related papers

Related papers: CausalKANs: interpretable treatment effect estimat…

200 papers

Meta-analysis is commonly used to combine results from multiple clinical trials, but traditional meta-analysis methods do not refer explicitly to a population of individuals to whom the results apply and it is not clear how to use their…

Kolmogorov-Arnold Networks (KANs) have inspired numerous works exploring their applications across a wide range of scientific problems, with the potential to replace Multilayer Perceptrons (MLPs). While many KANs are designed using basis…

Machine Learning · Computer Science 2025-03-11 Hoang-Thang Ta , Anh Tran

Physics-informed neural networks have proven to be a powerful tool for solving differential equations, leveraging the principles of physics to inform the learning process. However, traditional deep neural networks often face challenges in…

Machine Learning · Computer Science 2024-07-30 Subhajit Patra , Sonali Panda , Bikram Keshari Parida , Mahima Arya , Kurt Jacobs , Denys I. Bondar , Abhijit Sen

Interpretability has become a necessary feature for machine learning models deployed in critical scenarios, e.g. legal system, healthcare. In these situations, algorithmic decisions may have (potentially negative) long-lasting effects on…

Machine Learning · Computer Science 2021-12-21 An-phi Nguyen , Maria Rodriguez Martinez

Kolmogorov-Arnold Networks (KANs) are a class of neural networks that have received increased attention in recent literature. In contrast to MLPs, KANs leverage parameterized, trainable activation functions and offer several benefits…

Machine Learning · Computer Science 2025-11-14 Jamison Moody , James Usevitch

Within the field of causal inference, we consider the problem of estimating heterogeneous treatment effects from data. We propose and validate a novel approach for learning feature representations to aid the estimation of the conditional…

Machine Learning · Statistics 2022-06-23 Michael C. Burkhart , Gabriel Ruiz

We address the estimation of conditional average treatment effects (CATEs) for structured treatments (e.g., graphs, images, texts). Given a weak condition on the effect, we propose the generalized Robinson decomposition, which (i) isolates…

Machine Learning · Computer Science 2021-10-29 Jean Kaddour , Yuchen Zhu , Qi Liu , Matt J. Kusner , Ricardo Silva

Heart failure is a leading cause of global mortality, necessitating improved diagnostic strategies. Classical machine learning models struggle with challenges such as high-dimensional data, class imbalances, poor feature representations,…

Machine Learning · Computer Science 2025-08-19 Md Abrar Jahin , Md. Akmol Masud , M. F. Mridha , Zeyar Aung , Nilanjan Dey

Effective and interpretable classification of medical images is a challenge in computer-aided diagnosis, especially in resource-limited clinical settings. This study introduces spline-based Kolmogorov-Arnold Networks (KANs) for accurate…

Computer Vision and Pattern Recognition · Computer Science 2025-09-18 Kaniz Fatema , Emad A. Mohammed , Sukhjit Singh Sehra

Causal inference is capable of estimating the treatment effect (i.e., the causal effect of treatment on the outcome) to benefit the decision making in various domains. One fundamental challenge in this research is that the treatment…

Machine Learning · Computer Science 2021-12-28 Qian Li , Zhichao Wang , Shaowu Liu , Gang Li , Guandong Xu

Physics-Informed Neural Networks (PINNs) have emerged as a powerful mesh-free framework for solving ordinary and partial differential equations by embedding the governing physical laws directly into the loss function. However, their…

Neural density estimation has seen widespread applications in the gravitational-wave (GW) data analysis, which enables real-time parameter estimation for compact binary coalescences and enhances rapid inference for subsequent analysis such…

General Relativity and Quantum Cosmology · Physics 2025-12-05 Wenshuai Liu , Yiming Dong , Ziming Wang , Lijing Shao

Understanding and inferencing Heterogeneous Treatment Effects (HTE) and Conditional Average Treatment Effects (CATE) are vital for developing personalized treatment recommendations. Many state-of-the-art approaches achieve inspiring…

Machine Learning · Computer Science 2024-08-28 Chan Hsu , Jun-Ting Wu , Yihuang Kang

Graph Representation Learning aims to create effective embeddings for nodes and edges that encapsulate their features and relationships. Graph Neural Networks (GNNs) leverage neural networks to model complex graph structures. Recently, the…

Machine Learning · Computer Science 2025-01-23 Muhieddine Shebaro , Jelena Tešić

Hypergraph representation learning has garnered increasing attention across various domains due to its capability to model high-order relationships. Traditional methods often rely on hypergraph neural networks (HNNs) employing message…

Machine Learning · Computer Science 2025-03-18 Xiangfei Fang , Boying Wang , Chengying Huan , Shaonan Ma , Heng Zhang , Chen Zhao

Mechanistic interpretability aims to reverse engineer neural networks by uncovering which high-level algorithms they implement. Causal abstraction provides a precise notion of when a network implements an algorithm, i.e., a causal model of…

Machine Learning · Computer Science 2025-03-17 Theodora-Mara Pîslar , Sara Magliacane , Atticus Geiger

Existing low-light image enhancement methods are difficult to fit the complex nonlinear relationship between normal and low-light images due to uneven illumination and noise effects. The recently proposed Kolmogorov-Arnold networks (KANs)…

Computer Vision and Pattern Recognition · Computer Science 2025-02-07 Aoxiang Ning , Minglong Xue , Jinhong He , Chengyun Song

The Kolmogorov-Arnold representation theorem offers a theoretical alternative to Multi-Layer Perceptrons (MLPs) by placing learnable univariate functions on edges rather than nodes. While recent implementations such as Kolmogorov-Arnold…

Machine Learning · Computer Science 2026-01-28 Oscar Eliasson

Kolmogorov-Arnold Networks (KAN) has recently attracted significant attention as a promising alternative to traditional Multi-Layer Perceptrons (MLP). Despite their theoretical appeal, KAN require validation on large-scale benchmark…

Machine Learning · Computer Science 2024-09-12 Chang Dong , Liangwei Zheng , Weitong Chen

Graph Neural Networks (GNNs) have shown strong performance on graph-structured data, but their reliance on graph connectivity often limits scalability and efficiency. Kolmogorov-Arnold Networks (KANs), a recent architecture with learnable…

Machine Learning · Computer Science 2025-08-12 Yuan-Hung Chao , Chia-Hsun Lu , Chih-Ya Shen