Machine Learning

Authors and titles for January 2025

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

Title: Highly Optimized Kernels and Fine-Grained Codebooks for LLM Inference on Arm CPUs

Dibakar Gope, David Mansell, Danny Loh, Ian Bratt

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

[2] arXiv:2501.00042 [pdf, other]

Title: Resource-Efficient Transformer Architecture: Optimizing Memory and Execution Time for Real-Time Applications

Krisvarish V, Priyadarshini T, K P Abhishek Sri Saai, Vaidehi Vijayakumar

Comments: 5 pages, 1 figure

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

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

Title: Numerical solutions of fixed points in two-dimensional Kuramoto-Sivashinsky equation expedited by reinforcement learning

Juncheng Jiang, Dongdong Wan, Mengqi Zhang

Subjects: Machine Learning (cs.LG)

[4] arXiv:2501.00048 [pdf, other]

Title: Stroke Prediction using Clinical and Social Features in Machine Learning

Aidan Chadha

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

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

Title: DDD-GenDT: Dynamic Data-driven Generative Digital Twin Framework

Yu-Zheng Lin, Qinxuan Shi, Zhanglong Yang, Banafsheh Saber Latibari, Sicong Shao, Soheil Salehi, Pratik Satam

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

[6] arXiv:2501.00052 [pdf, html, other]

Title: Efficient and Scalable Deep Reinforcement Learning for Mean Field Control Games

Nianli Peng, Yilin Wang

Subjects: Machine Learning (cs.LG); Computer Science and Game Theory (cs.GT); Multiagent Systems (cs.MA)

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

Title: AdvAnchor: Enhancing Diffusion Model Unlearning with Adversarial Anchors

Mengnan Zhao, Lihe Zhang, Xingyi Yang, Tianhang Zheng, Baocai Yin

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

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

Title: Transforming CCTV cameras into NO$_2$ sensors at city scale for adaptive policymaking

Mohamed R. Ibrahim, Terry Lyons

Comments: 43 pages

Subjects: Machine Learning (cs.LG); Artificial Intelligence (cs.AI); Computers and Society (cs.CY)

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

Title: VisTabNet: Adapting Vision Transformers for Tabular Data

Witold Wydmański, Ulvi Movsum-zada, Jacek Tabor, Marek Śmieja

Subjects: Machine Learning (cs.LG); Artificial Intelligence (cs.AI); Computer Vision and Pattern Recognition (cs.CV)

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

Title: Training-free Heterogeneous Model Merging

Zhengqi Xu, Han Zheng, Jie Song, Li Sun, Mingli Song

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

[11] arXiv:2501.00063 [pdf, html, other]

Title: "Generative Models for Financial Time Series Data: Enhancing Signal-to-Noise Ratio and Addressing Data Scarcity in A-Share Market

Guangming Che

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

[12] arXiv:2501.00065 [pdf, other]

Title: Predicting Preschoolers' Externalizing Problems with Mother-Child Interaction Dynamics and Deep Learning

Xi Chen, Yu Ji, Cong Xia, Wen Wu

Comments: 34 pages, 3 figures, 2 tables

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

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

Title: Open-Book Neural Algorithmic Reasoning

Hefei Li, Chao Peng, Chenyang Xu, Zhengfeng Yang

Comments: Appeared at NeurIPS 2024

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

[14] arXiv:2501.00076 [pdf, other]

Title: A Novel Framework for Learning Stochastic Representations for Sequence Generation and Recognition

Jungsik Hwang, Ahmadreza Ahmadi

Comments: 14 pages, 6 figures

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

[15] arXiv:2501.00085 [pdf, other]

Title: Machine Learning-Based Security Policy Analysis

Krish Jain, Joann Sum, Pranav Kapoor, Amir Eaman

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

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

Title: An Unsupervised Anomaly Detection in Electricity Consumption Using Reinforcement Learning and Time Series Forest Based Framework

Jihan Ghanim, Mariette Awad

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

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

Title: Urban Water Consumption Forecasting Using Deep Learning and Correlated District Metered Areas

Kleanthis Malialis, Nefeli Mavri, Stelios G. Vrachimis, Marios S. Kyriakou, Demetrios G. Eliades, Marios M. Polycarpou

Comments: Keywords: urban water management, water consumption, time series forecasting

Subjects: Machine Learning (cs.LG); Computers and Society (cs.CY)

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

Title: Class-based Subset Selection for Transfer Learning under Extreme Label Shift

Akul Goyal, Carl Edwards

Comments: 19 pages

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

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

Title: Federated Learning with Workload Reduction through Partial Training of Client Models and Entropy-Based Data Selection

Hongrui Shi, Valentin Radu, Po Yang

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

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

Title: TrajLearn: Trajectory Prediction Learning using Deep Generative Models

Amirhossein Nadiri, Jing Li, Ali Faraji, Ghadeer Abuoda, Manos Papagelis

Subjects: Machine Learning (cs.LG); Computer Vision and Pattern Recognition (cs.CV); Robotics (cs.RO)

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

Title: SepsisCalc: Integrating Clinical Calculators into Early Sepsis Prediction via Dynamic Temporal Graph Construction

Changchang Yin, Shihan Fu, Bingsheng Yao, Thai-Hoang Pham, Weidan Cao, Dakuo Wang, Jeffrey Caterino, Ping Zhang

Subjects: Machine Learning (cs.LG); Artificial Intelligence (cs.AI); Human-Computer Interaction (cs.HC)

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

Title: Towards Unraveling and Improving Generalization in World Models

Qiaoyi Fang, Weiyu Du, Hang Wang, Junshan Zhang

Comments: An earlier version of this paper was submitted to NeurIPS and received ratings of (7, 6, 6). The reviewers' comments and the original draft are available at OpenReview. This version contains minor modifications based on that submission

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

[23] arXiv:2501.00200 [pdf, html, other]

Title: Scalable Neural Network Verification with Branch-and-bound Inferred Cutting Planes

Duo Zhou, Christopher Brix, Grani A Hanasusanto, Huan Zhang

Comments: Accepted by NeurIPS 2024. BICCOS is part of the alpha-beta-CROWN verifier, the VNN-COMP 2024 winner

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

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

Title: Federated Deep Subspace Clustering

Yupei Zhang, Ruojia Feng, Yifei Wang, Xuequn Shang

Comments: 8pages,4 figures, 4 Tables

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

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

Title: Towards Pattern-aware Data Augmentation for Temporal Knowledge Graph Completion

Jiasheng Zhang, Deqiang Ouyang, Shuang Liang, Jie Shao

Subjects: Machine Learning (cs.LG); Databases (cs.DB); Information Retrieval (cs.IR)

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

Title: Outlier-Robust Training of Machine Learning Models

Rajat Talak, Charis Georgiou, Jingnan Shi, Luca Carlone

Subjects: Machine Learning (cs.LG); Computer Vision and Pattern Recognition (cs.CV)

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

Title: ReFormer: Generating Radio Fakes for Data Augmentation

Yagna Kaasaragadda, Silvija Kokalj-Filipovic

Subjects: Machine Learning (cs.LG); Signal Processing (eess.SP)

[28] arXiv:2501.00305 [pdf, other]

Title: diffIRM: A Diffusion-Augmented Invariant Risk Minimization Framework for Spatiotemporal Prediction over Graphs

Zhaobin Mo, Haotian Xiang, Xuan Di

Subjects: Machine Learning (cs.LG)

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

Title: Fast and Interpretable Mixed-Integer Linear Program Solving by Learning Model Reduction

Yixuan Li, Can Chen, Jiajun Li, Jiahui Duan, Xiongwei Han, Tao Zhong, Vincent Chau, Weiwei Wu, Wanyuan Wang

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

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

Title: A New Dataset and Methodology for Malicious URL Classification

Ilan Schvartzman, Roei Sarussi, Maor Ashkenazi, Ido kringel, Yaniv Tocker, Tal Furman Shohet

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

[31] arXiv:2501.00365 [pdf, html, other]

Title: Low-Rank Adaptation for Foundation Models: A Comprehensive Review

Menglin Yang, Jialin Chen, Yifei Zhang, Jiahong Liu, Jiasheng Zhang, Qiyao Ma, Harshit Verma, Qianru Zhang, Min Zhou, Irwin King, Rex Ying

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

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

Title: Federated Dropout: Convergence Analysis and Resource Allocation

Sijing Xie, Dingzhu Wen, Xiaonan Liu, Changsheng You, Tharmalingam Ratnarajah, Kaibin Huang

Subjects: Machine Learning (cs.LG); Information Theory (cs.IT)

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

Title: Toward Information Theoretic Active Inverse Reinforcement Learning

Ondrej Bajgar, Sid William Gould, Rohan Narayan Langford Mitta, Jonathon Liu, Oliver Newcombe, Jack Golden

Comments: NeurIPS 2024 Workshop on Bayesian Decision-making and Uncertainty

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

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

Title: Efficient Relational Context Perception for Knowledge Graph Completion

Wenkai Tu, Guojia Wan, Zhengchun Shang, Bo Du

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

[35] arXiv:2501.00418 [pdf, html, other]

Title: Generalizing Trust: Weak-to-Strong Trustworthiness in Language Models

Martin Pawelczyk, Lillian Sun, Zhenting Qi, Aounon Kumar, Himabindu Lakkaraju

Comments: The first two authors contributed equally

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

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

Title: KAE: Kolmogorov-Arnold Auto-Encoder for Representation Learning

Fangchen Yu, Ruilizhen Hu, Yidong Lin, Yuqi Ma, Zhenghao Huang, Wenye Li

Subjects: Machine Learning (cs.LG)

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

Title: Intuitive Analysis of the Quantization-based Optimization: From Stochastic and Quantum Mechanical Perspective

Jinwuk Seok, Changsik Cho

Comments: published in NeurIPS 2024 workshop OPT2024

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

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

Title: Differentiable Prompt Learning for Vision Language Models

Zhenhan Huang, Tejaswini Pedapati, Pin-Yu Chen, Jianxi Gao

Subjects: Machine Learning (cs.LG); Artificial Intelligence (cs.AI); Computation and Language (cs.CL); Computer Vision and Pattern Recognition (cs.CV)

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

Title: SAT-LDM: Provably Generalizable Image Watermarking for Latent Diffusion Models with Self-Augmented Training

Lu Zhang, Liang Zeng

Comments: 21 pages, 7 figures

Subjects: Machine Learning (cs.LG); Cryptography and Security (cs.CR); Computer Vision and Pattern Recognition (cs.CV)

[40] arXiv:2501.00464 [pdf, html, other]

Title: Addressing Challenges in Data Quality and Model Generalization for Malaria Detection

Kiswendsida Kisito Kabore, Desire Guel

Comments: 22 pages, 17 figures, 17 tables, In: Journal of Sensor Networks and Data Communications (JSNDC). ISSN: 2994-6433. DOI: https://doi.org/10.33140/JSNDC.04.03.09

Subjects: Machine Learning (cs.LG); Signal Processing (eess.SP)

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

Title: Dementia Detection using Multi-modal Methods on Audio Data

Saugat Kannojia, Anirudh Praveen, Danish Vasdev, Saket Nandedkar, Divyansh Mittal, Sarthak Kalankar, Shaurya Johari, Vipul Arora

Comments: 4 pages

Subjects: Machine Learning (cs.LG)

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

Title: Score-Based Metropolis-Hastings Algorithms

Ahmed Aloui, Ali Hasan, Juncheng Dong, Zihao Wu, Vahid Tarokh

Subjects: Machine Learning (cs.LG); Computation (stat.CO)

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

Title: Exploring Physics-Informed Neural Networks for Crop Yield Loss Forecasting

Miro Miranda, Marcela Charfuelan, Andreas Dengel

Comments: 6 pages, 2 figures, NeurIPS 2024 Workshop on Tackling Climate Change with Machine Learning

Journal-ref: https://www.climatechange.ai/papers/neurips2024/45

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

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

Title: Active Learning of General Halfspaces: Label Queries vs Membership Queries

Ilias Diakonikolas, Daniel M. Kane, Mingchen Ma

Comments: Accepted by NeurIPS 2024

Subjects: Machine Learning (cs.LG)

[45] arXiv:2501.00511 [pdf, other]

Title: Stochastic Extragradient with Flip-Flop Shuffling & Anchoring: Provable Improvements

Jiseok Chae, Chulhee Yun, Donghwan Kim

Comments: 73+7 pages, 4 figures. Published in NeurIPS 2024

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

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

Title: PyMilo: A Python Library for ML I/O

AmirHosein Rostami, Sepand Haghighi, Sadra Sabouri, Alireza Zolanvari

Comments: 7 pages, 5 figures, 2 tables, 3 code blocks

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

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

Title: Rapid Learning in Constrained Minimax Games with Negative Momentum

Zijian Fang, Zongkai Liu, Chao Yu, Chaohao Hu

Subjects: Machine Learning (cs.LG)

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

Title: Adaptive Tabu Dropout for Regularization of Deep Neural Network

Md. Tarek Hasan, Arifa Akter, Mohammad Nazmush Shamael, Md Al Emran Hossain, H. M. Mutasim Billah, Sumayra Islam, Swakkhar Shatabda

Journal-ref: Neural Information Processing, ICONIP 2022, Lecture Notes in Computer Science 13623, Springer Cham, 2023, 334-345

Subjects: Machine Learning (cs.LG)

[49] arXiv:2501.00555 [pdf, other]

Title: Monty Hall and Optimized Conformal Prediction to Improve Decision-Making with LLMs

Harit Vishwakarma, Alan Mishler, Thomas Cook, Niccolò Dalmasso, Natraj Raman, Sumitra Ganesh

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

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

Title: Unbiased GNN Learning via Fairness-Aware Subgraph Diffusion

Abdullah Alchihabi, Yuhong Guo

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