arXiv Machine Learning

Two-Level Softmax Sampling Done Right: Correcting Bias from Size Imbalance and Dispersion

The paper addresses the bias introduced by two‑level softmax (2LS) sampling, which partitions items into clusters and samples a cluster before an item. It identifies that 2LS misweights clusters by ignoring size imbalance and intra‑cluster similarity dispersion, leading to systematic sampling bias. The authors propose two corrected methods, Size‑Corrected 2LS (S‑2LS) and Size‑ and Dispersion‑Corrected 2LS (SD‑2LS), that eliminate these biases with negligible computational overhead and demonstrate improved sampling performance on five large‑scale datasets.

arXiv Machine Learning
Oct 7

Fed-BRDECS: Privacy-Preserving and Heterogeneity-Aware Federated Deep Embedded Clustering

Fed-BRDECS is a federated deep embedded clustering framework that protects client privacy by replacing globally normalized clustering objectives with a locally computable sample‑stability loss, eliminating the need to share sensitive soft‑assignment distributions. It addresses non‑IID client data through prediction‑balanced sampling, which oversamples rare predicted clusters, and centroid‑level restarting, which refreshes biased or inactive centroids. Experiments on image, text, and time‑series datasets demonstrate that Fed‑BRDECS outperforms existing federated and deep clustering methods while adding no inference‑time cost for anomaly detection.

By Haemin Park, Diego Klabjan, Martin W. Braun, Xiuqi Li, Balakrishnan Ananthanarayanan
Hugging Face Trending Papers
Oct 29

CBE: A Complexity-adaptive Boundary-aware Ensemble Method for Imbalanced Data

The paper introduces CBE, a complexity‑adaptive boundary‑aware ensemble method designed to handle class imbalance in classification tasks. It links sampling weights to data complexity, dynamically adjusting boundary preference during under‑sampling, and employs a multi‑level complexity measure to categorize datasets into Easy, Median, and Hard. Experiments on 24 UCI and KEEL datasets show that CBE improves F‑measure and G‑mean on Easy and Median cases without harming performance on Hard cases.

arXiv AI
Oct 26

Internal Discrepancy for Contrastive Learning: Generalization Bounds and Downstream Behaviors

The paper introduces the concept of internal discrepancy, measuring the gap between the mean representation of positive samples and latent cluster centers in contrastive learning. It derives a generalization bound for downstream supervised tasks that depends on this discrepancy, providing a unifying framework for existing assumptions and motivating a relaxed k‑means loss to reduce it. The study also explores how contrastive learning affects downstream behavior, particularly in class‑imbalanced settings, and establishes properties of the resulting supervised losses.

By Jialiang Wang, Qing Ma, Xiong Zhou, Gangfeng Hu, Haoliang Li, Xianming Liu