Machine Learning

Authors and titles for June 2024

[1] arXiv:2406.00061 [pdf, html, other]

Title: STAT: Shrinking Transformers After Training

Megan Flynn, Alexander Wang, Dean Edward Alvarez, Christopher De Sa, Anil Damle

Subjects: Machine Learning (cs.LG); Artificial Intelligence (cs.AI); Computation and Language (cs.CL)

[2] arXiv:2406.00073 [pdf, html, other]

Title: A Novel Review of Stability Techniques for Improved Privacy-Preserving Machine Learning

Coleman DuPlessie, Aidan Gao

Subjects: Machine Learning (cs.LG); Cryptography and Security (cs.CR)

[3] arXiv:2406.00075 [pdf, html, other]

Title: Arbitrary-Length Generalization for Addition in a Tiny Transformer

Alexandre Galvao Patriota

Comments: Testing-Digits.R output with 50-digit numbers (8 pages, 1 figure)

Subjects: Machine Learning (cs.LG); Applications (stat.AP); Machine Learning (stat.ML)

[4] arXiv:2406.00079 [pdf, html, other]

Title: Decision Mamba: Reinforcement Learning via Hybrid Selective Sequence Modeling

Sili Huang, Jifeng Hu, Zhejian Yang, Liwei Yang, Tao Luo, Hechang Chen, Lichao Sun, Bo Yang

Comments: arXiv admin note: text overlap with arXiv:2405.20692. arXiv admin note: text overlap with arXiv:2405.20692; text overlap with arXiv:2305.16554, arXiv:2210.14215 by other authors

Subjects: Machine Learning (cs.LG)

[5] arXiv:2406.00080 [pdf, html, other]

Title: An Efficient Multi Quantile Regression Network with Ad Hoc Prevention of Quantile Crossing

Jens Decke, Arne Jenß, Bernhard Sick, Christian Gruhl

Subjects: Machine Learning (cs.LG); Machine Learning (stat.ML)

[6] arXiv:2406.00081 [pdf, other]

Title: From Structured to Unstructured:A Comparative Analysis of Computer Vision and Graph Models in solving Mesh-based PDEs

Jens Decke, Olaf Wünsch, Bernhard Sick, Christian Gruhl

Subjects: Machine Learning (cs.LG); Computational Engineering, Finance, and Science (cs.CE); Computer Vision and Pattern Recognition (cs.CV)

[7] arXiv:2406.00104 [pdf, html, other]

Title: Scalable Bayesian Learning with posteriors

Samuel Duffield, Kaelan Donatella, Johnathan Chiu, Phoebe Klett, Daniel Simpson

Journal-ref: Published as a conference paper at ICLR 2025

Subjects: Machine Learning (cs.LG); Machine Learning (stat.ML)

[8] arXiv:2406.00118 [pdf, html, other]

Title: ADEP: A Novel Approach Based on Discriminator-Enhanced Encoder-Decoder Architecture for Accurate Prediction of Adverse Effects in Polypharmacy

Katayoun Kobraei, Mehrdad Baradaran, Seyed Mohsen Sadeghi, Raziyeh Masumshah, Changiz Eslahchi

Comments: 13 pages, 1 figure

Subjects: Machine Learning (cs.LG); Quantitative Methods (q-bio.QM)

[9] arXiv:2406.00120 [pdf, html, other]

Title: Reward Machines for Deep RL in Noisy and Uncertain Environments

Andrew C. Li, Zizhao Chen, Toryn Q. Klassen, Pashootan Vaezipoor, Rodrigo Toro Icarte, Sheila A. McIlraith

Subjects: Machine Learning (cs.LG); Artificial Intelligence (cs.AI); Formal Languages and Automata Theory (cs.FL)

[10] arXiv:2406.00131 [pdf, html, other]

Title: From Unstructured Data to In-Context Learning: Exploring What Tasks Can Be Learned and When

Kevin Christian Wibisono, Yixin Wang

Comments: 39 pages

Subjects: Machine Learning (cs.LG); Computation and Language (cs.CL); Machine Learning (stat.ML)

[11] arXiv:2406.00132 [pdf, other]

Title: QuanTA: Efficient High-Rank Fine-Tuning of LLMs with Quantum-Informed Tensor Adaptation

Zhuo Chen, Rumen Dangovski, Charlotte Loh, Owen Dugan, Di Luo, Marin Soljačić

Subjects: Machine Learning (cs.LG); Quantum Physics (quant-ph)

[12] arXiv:2406.00133 [pdf, html, other]

Title: Streamflow Prediction with Uncertainty Quantification for Water Management: A Constrained Reasoning and Learning Approach

Mohammed Amine Gharsallaoui, Bhupinderjeet Singh, Supriya Savalkar, Aryan Deshwal, Yan Yan, Ananth Kalyanaraman, Kirti Rajagopalan, Janardhan Rao Doppa

Subjects: Machine Learning (cs.LG); Artificial Intelligence (cs.AI)

[13] arXiv:2406.00134 [pdf, html, other]

Title: Anomaly Detection in Dynamic Graphs: A Comprehensive Survey

Ocheme Anthony Ekle, William Eberle

Comments: 32 pages (double column), 4 figures, and the manuscript has just been accepted in ACM Journals of Transactions on Knowledge Discovery from Data (TKDD)

Subjects: Machine Learning (cs.LG); Artificial Intelligence (cs.AI)

[14] arXiv:2406.00144 [pdf, html, other]

Title: Query2CAD: Generating CAD models using natural language queries

Akshay Badagabettu, Sai Sravan Yarlagadda, Amir Barati Farimani

Comments: 8 pages, 5 figures

Subjects: Machine Learning (cs.LG); Artificial Intelligence (cs.AI); Computational Engineering, Finance, and Science (cs.CE)

[15] arXiv:2406.00150 [pdf, html, other]

Title: Non-Federated Multi-Task Split Learning for Heterogeneous Sources

Yilin Zheng, Atilla Eryilmaz

Subjects: Machine Learning (cs.LG); Distributed, Parallel, and Cluster Computing (cs.DC)

[16] arXiv:2406.00153 [pdf, html, other]

Title: $μ$LO: Compute-Efficient Meta-Generalization of Learned Optimizers

Benjamin Thérien, Charles-Étienne Joseph, Boris Knyazev, Edouard Oyallon, Irina Rish, Eugene Belilovsky

Subjects: Machine Learning (cs.LG)

[17] arXiv:2406.00177 [pdf, html, other]

Title: Flexible and Efficient Surrogate Gradient Modeling with Forward Gradient Injection

Sebastian Otte

Subjects: Machine Learning (cs.LG); Neural and Evolutionary Computing (cs.NE)

[18] arXiv:2406.00209 [pdf, html, other]

Title: Mamba State-Space Models Are Lyapunov-Stable Learners

John T. Halloran, Manbir Gulati, Paul F. Roysdon

Comments: 18 pages, 4 figure panels, 5 tables

Subjects: Machine Learning (cs.LG); Artificial Intelligence (cs.AI)

[19] arXiv:2406.00234 [pdf, html, other]

Title: Learning to Stabilize Unknown LTI Systems on a Single Trajectory under Stochastic Noise

Ziyi Zhang, Yorie Nakahira, Guannan Qu

Subjects: Machine Learning (cs.LG); Systems and Control (eess.SY)

[20] arXiv:2406.00240 [pdf, html, other]

Title: Exploring Vulnerabilities and Protections in Large Language Models: A Survey

Frank Weizhen Liu, Chenhui Hu

Subjects: Machine Learning (cs.LG); Computation and Language (cs.CL); Cryptography and Security (cs.CR)

[21] arXiv:2406.00249 [pdf, html, other]

Title: Privacy Challenges in Meta-Learning: An Investigation on Model-Agnostic Meta-Learning

Mina Rafiei, Mohammadmahdi Maheri, Hamid R. Rabiee

Subjects: Machine Learning (cs.LG); Cryptography and Security (cs.CR)

[22] arXiv:2406.00262 [pdf, html, other]

Title: Contrastive Learning Via Equivariant Representation

Sifan Song, Jinfeng Wang, Qiaochu Zhao, Xiang Li, Dufan Wu, Angelos Stefanidis, Jionglong Su, S. Kevin Zhou, Quanzheng Li

Comments: Preprint. Under review

Subjects: Machine Learning (cs.LG); Artificial Intelligence (cs.AI)

[23] arXiv:2406.00276 [pdf, other]

Title: Non-destructive Degradation Pattern Decoupling for Ultra-early Battery Prototype Verification Using Physics-informed Machine Learning

Shengyu Tao, Mengtian Zhang, Zixi Zhao, Haoyang Li, Ruifei Ma, Yunhong Che, Xin Sun, Lin Su, Xiangyu Chen, Zihao Zhou, Heng Chang, Tingwei Cao, Xiao Xiao, Yaojun Liu, Wenjun Yu, Zhongling Xu, Yang Li, Han Hao, Xuan Zhang, Xiaosong Hu, Guangmin ZHou

Journal-ref: Energy Environ. Sci., 2025,18, 1544-1559

Subjects: Machine Learning (cs.LG); Artificial Intelligence (cs.AI); Computational Engineering, Finance, and Science (cs.CE); Data Analysis, Statistics and Probability (physics.data-an)

[24] arXiv:2406.00281 [pdf, html, other]

Title: Cross-Table Pretraining towards a Universal Function Space for Heterogeneous Tabular Data

Jintai Chen, Zhen Lin, Qiyuan Chen, Jimeng Sun

Subjects: Machine Learning (cs.LG); Artificial Intelligence (cs.AI)

[25] arXiv:2406.00288 [pdf, html, other]

Title: Neural Optimal Transport with Lagrangian Costs

Aram-Alexandre Pooladian, Carles Domingo-Enrich, Ricky T. Q. Chen, Brandon Amos

Comments: UAI 2024

Subjects: Machine Learning (cs.LG); Machine Learning (stat.ML)

[26] arXiv:2406.00291 [pdf, html, other]

Title: Multi-Objective Neural Architecture Search by Learning Search Space Partitions

Yiyang Zhao, Linnan Wang, Tian Guo

Journal-ref: Journal of Machine Learning Research 25 (2024) 1-41

Subjects: Machine Learning (cs.LG); Artificial Intelligence (cs.AI)

[27] arXiv:2406.00300 [pdf, html, other]

Title: Coded Computing for Resilient Distributed Computing: A Learning-Theoretic Framework

Parsa Moradi, Behrooz Tahmasebi, Mohammad Ali Maddah-Ali

Comments: 35 pages, 7 figures

Journal-ref: 38th Conference on Neural Information Processing Systems (NeurIPS 2024)

Subjects: Machine Learning (cs.LG); Distributed, Parallel, and Cluster Computing (cs.DC); Information Theory (cs.IT)

[28] arXiv:2406.00302 [pdf, html, other]

Title: FedAST: Federated Asynchronous Simultaneous Training

Baris Askin, Pranay Sharma, Carlee Joe-Wong, Gauri Joshi

Comments: Accepted to UAI 2024

Subjects: Machine Learning (cs.LG); Distributed, Parallel, and Cluster Computing (cs.DC)

[29] arXiv:2406.00318 [pdf, html, other]

Title: KGLink: A column type annotation method that combines knowledge graph and pre-trained language model

Yubo Wang, Hao Xin, Lei Chen

Comments: To be published in ICDE 2024

Subjects: Machine Learning (cs.LG); Computation and Language (cs.CL); Information Retrieval (cs.IR)

[30] arXiv:2406.00324 [pdf, html, other]

Title: Do's and Don'ts: Learning Desirable Skills with Instruction Videos

Hyunseung Kim, Byungkun Lee, Hojoon Lee, Dongyoon Hwang, Donghu Kim, Jaegul Choo

Comments: published at NeurIPS 2024

Subjects: Machine Learning (cs.LG); Artificial Intelligence (cs.AI)

[31] arXiv:2406.00332 [pdf, other]

Title: A Structured Review of Literature on Uncertainty in Machine Learning & Deep Learning

Fahimeh Fakour, Ali Mosleh, Ramin Ramezani

Subjects: Machine Learning (cs.LG); Artificial Intelligence (cs.AI); Machine Learning (stat.ML)

[32] arXiv:2406.00335 [pdf, html, other]

Title: Benchmarking for Deep Uplift Modeling in Online Marketing

Dugang Liu, Xing Tang, Yang Qiao, Miao Liu, Zexu Sun, Xiuqiang He, Zhong Ming

Subjects: Machine Learning (cs.LG)

[33] arXiv:2406.00368 [pdf, html, other]

Title: Learning Spatiotemporal Dynamical Systems from Point Process Observations

Valerii Iakovlev, Harri Lähdesmäki

Subjects: Machine Learning (cs.LG); Machine Learning (stat.ML)

[34] arXiv:2406.00371 [pdf, html, other]

Title: Alternative Methods to SHAP Derived from Properties of Kernels: A Note on Theoretical Analysis

Kazuhiro Hiraki, Shinichi Ishihara, Junnosuke Shino

Subjects: Machine Learning (cs.LG)

[35] arXiv:2406.00394 [pdf, other]

Title: Learning Causal Abstractions of Linear Structural Causal Models

Riccardo Massidda, Sara Magliacane, Davide Bacciu

Subjects: Machine Learning (cs.LG); Artificial Intelligence (cs.AI); Methodology (stat.ME)

[36] arXiv:2406.00396 [pdf, html, other]

Title: Stochastic Resetting Mitigates Latent Gradient Bias of SGD from Label Noise

Youngkyoung Bae, Yeongwoo Song, Hawoong Jeong

Comments: 30 pages, 14 figures

Journal-ref: Mach. Learn.: Sci. Technol. 6 (2025) 015062

Subjects: Machine Learning (cs.LG); Statistical Mechanics (cond-mat.stat-mech); Artificial Intelligence (cs.AI); Machine Learning (stat.ML)

[37] arXiv:2406.00403 [pdf, html, other]

Title: Dual-perspective Cross Contrastive Learning in Graph Transformers

Zelin Yao, Chuang Liu, Xueqi Ma, Mukun Chen, Jia Wu, Xiantao Cai, Bo Du, Wenbin Hu

Comments: 12 pages, 5 figures, submitted to IEEE TKDE

Subjects: Machine Learning (cs.LG); Artificial Intelligence (cs.AI)

[38] arXiv:2406.00410 [pdf, html, other]

Title: Posterior Label Smoothing for Node Classification

Jaeseung Heo, Moonjeong Park, Dongwoo Kim

Subjects: Machine Learning (cs.LG); Artificial Intelligence (cs.AI)

[39] arXiv:2406.00418 [pdf, html, other]

Title: GATE: How to Keep Out Intrusive Neighbors

Nimrah Mustafa, Rebekka Burkholz

Comments: 26 pages. Published at the International Conference on Machine Learning (ICML), 2024

Subjects: Machine Learning (cs.LG)

[40] arXiv:2406.00426 [pdf, other]

Title: InterpreTabNet: Distilling Predictive Signals from Tabular Data by Salient Feature Interpretation

Jacob Si, Wendy Yusi Cheng, Michael Cooper, Rahul G. Krishnan

Comments: ICML 2024 Spotlight

Subjects: Machine Learning (cs.LG)

[41] arXiv:2406.00431 [pdf, html, other]

Title: SpaFL: Communication-Efficient Federated Learning with Sparse Models and Low computational Overhead

Minsu Kim, Walid Saad, Merouane Debbah, Choong Seon Hong

Comments: Published in NeurIPS 2024

Subjects: Machine Learning (cs.LG); Artificial Intelligence (cs.AI); Distributed, Parallel, and Cluster Computing (cs.DC)

[42] arXiv:2406.00438 [pdf, html, other]

Title: Stein Random Feature Regression

Houston Warren, Rafael Oliveira, Fabio Ramos

Comments: To appear at UAI24

Subjects: Machine Learning (cs.LG); Machine Learning (stat.ML)

[43] arXiv:2406.00452 [pdf, html, other]

Title: Towards a Unified Framework of Clustering-based Anomaly Detection

Zeyu Fang, Ming Gu, Sheng Zhou, Jiawei Chen, Qiaoyu Tan, Haishuai Wang, Jiajun Bu

Subjects: Machine Learning (cs.LG); Artificial Intelligence (cs.AI)

[44] arXiv:2406.00456 [pdf, html, other]

Title: Mix-of-Granularity: Optimize the Chunking Granularity for Retrieval-Augmented Generation

Zijie Zhong, Hanwen Liu, Xiaoya Cui, Xiaofan Zhang, Zengchang Qin

Comments: COLING 2025 conference paper. 19 pages, 6 figures and 11 tables

Subjects: Machine Learning (cs.LG); Artificial Intelligence (cs.AI); Computation and Language (cs.CL)

[45] arXiv:2406.00469 [pdf, html, other]

Title: Learning to Solve Multiresolution Matrix Factorization by Manifold Optimization and Evolutionary Metaheuristics

Truong Son Hy, Thieu Khang, Risi Kondor

Comments: arXiv admin note: substantial text overlap with arXiv:2111.01940

Subjects: Machine Learning (cs.LG); Neural and Evolutionary Computing (cs.NE)

[46] arXiv:2406.00483 [pdf, html, other]

Title: Exploring the limits of Hierarchical World Models in Reinforcement Learning

Robin Schiewer, Anand Subramoney, Laurenz Wiskott

Comments: 26 pages, 14 figures

Subjects: Machine Learning (cs.LG)

[47] arXiv:2406.00487 [pdf, html, other]

Title: Optimistic Rates for Learning from Label Proportions

Gene Li, Lin Chen, Adel Javanmard, Vahab Mirrokni

Comments: Accepted to COLT 2024. Comments welcome

Subjects: Machine Learning (cs.LG); Artificial Intelligence (cs.AI); Machine Learning (stat.ML)

[48] arXiv:2406.00488 [pdf, html, other]

Title: Federated Model Heterogeneous Matryoshka Representation Learning

Liping Yi, Han Yu, Chao Ren, Gang Wang, Xiaoguang Liu, Xiaoxiao Li

Subjects: Machine Learning (cs.LG); Distributed, Parallel, and Cluster Computing (cs.DC)

[49] arXiv:2406.00489 [pdf, html, other]

Title: Efficient Sign-Based Optimization: Accelerating Convergence via Variance Reduction

Wei Jiang, Sifan Yang, Wenhao Yang, Lijun Zhang

Subjects: Machine Learning (cs.LG); Optimization and Control (math.OC)

[50] arXiv:2406.00494 [pdf, html, other]

Title: Activation-Descent Regularization for Input Optimization of ReLU Networks

Hongzhan Yu, Sicun Gao

Comments: ICML'24 Proceedings

Subjects: Machine Learning (cs.LG); Artificial Intelligence (cs.AI)