Using sliced mutual information to study memorization and generalization in deep neural networks

S Wongso, R Ghosh, M Motani - International Conference on …, 2023 - proceedings.mlr.press
In this paper, we study the memorization and generalization behaviour of deep neural
networks (DNNs) using sliced mutual information (SMI), which is the average of the mutual …

On Neural Networks Fitting, Compression, and Generalization Behavior via Information-Bottleneck-like Approaches

Z Lyu, G Aminian, MRD Rodrigues - Entropy, 2023 - mdpi.com
It is well-known that a neural network learning process—along with its connections to fitting,
compression, and generalization—is not yet well understood. In this paper, we propose a …

A lightweight and personalized edge federated learning model

P Yuan, L Shi, X Zhao, J Zhang - Complex & Intelligent Systems, 2024 - Springer
As a new distributed machine learning paradigm, federated learning has gained increasing
attention in the industry and research community. However, federated learning is …

Pointwise Sliced Mutual Information for Neural Network Explainability

S Wongso, R Ghosh, M Motani - 2023 IEEE International …, 2023 - ieeexplore.ieee.org
When deploying deep learning models such as convolutional neural networks (CNNs) in
safety-critical domains, it is important to understand the predictions made by these black-box …

Gone With the Bits: Benchmarking Bias in Facial Phenotype Degradation Under Low-Rate Neural Compression

T Qiu, A Nichani, R Tadayon, H Jeong - ICML 2024 Next Generation of AI … - openreview.net
In this study, we investigate how facial phenotypes are distorted under neural image
compression and the disparity of this distortion across racial groups. Neural compression …