Towards a Fairer Non-negative Matrix Factorization

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Kassab, Lara, George, Erin, Needell, Deanna, Geng, Haowen, Nia, Nika Jafar, Li, Aoxi
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866917314743500800
author Kassab, Lara
George, Erin
Needell, Deanna
Geng, Haowen
Nia, Nika Jafar
Li, Aoxi
author_facet Kassab, Lara
George, Erin
Needell, Deanna
Geng, Haowen
Nia, Nika Jafar
Li, Aoxi
contents There has been a recent critical need to study fairness and bias in machine learning (ML) algorithms. Since there is clearly no one-size-fits-all solution to fairness, ML methods should be developed alongside bias mitigation strategies that are practical and approachable to the practitioner. Motivated by recent work on ``fair" PCA, here we consider the more challenging method of non-negative matrix factorization (NMF) as both a showcasing example and a method that is important in its own right for both topic modeling tasks and feature extraction for other ML tasks. We demonstrate that a modification of the objective function, by using a min-max formulation, may \textit{sometimes} be able to offer an improvement in fairness for groups in the population. We derive two methods for the objective minimization, a multiplicative update rule as well as an alternating minimization scheme, and discuss implementation practicalities. We include a suite of synthetic and real experiments that show how the method may improve fairness while also highlighting the important fact that this may sometime increase error for some individuals and fairness is not a rigid definition and method choice should strongly depend on the application at hand.
format Preprint
id arxiv_https___arxiv_org_abs_2411_09847
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Towards a Fairer Non-negative Matrix Factorization
Kassab, Lara
George, Erin
Needell, Deanna
Geng, Haowen
Nia, Nika Jafar
Li, Aoxi
Machine Learning
There has been a recent critical need to study fairness and bias in machine learning (ML) algorithms. Since there is clearly no one-size-fits-all solution to fairness, ML methods should be developed alongside bias mitigation strategies that are practical and approachable to the practitioner. Motivated by recent work on ``fair" PCA, here we consider the more challenging method of non-negative matrix factorization (NMF) as both a showcasing example and a method that is important in its own right for both topic modeling tasks and feature extraction for other ML tasks. We demonstrate that a modification of the objective function, by using a min-max formulation, may \textit{sometimes} be able to offer an improvement in fairness for groups in the population. We derive two methods for the objective minimization, a multiplicative update rule as well as an alternating minimization scheme, and discuss implementation practicalities. We include a suite of synthetic and real experiments that show how the method may improve fairness while also highlighting the important fact that this may sometime increase error for some individuals and fairness is not a rigid definition and method choice should strongly depend on the application at hand.
title Towards a Fairer Non-negative Matrix Factorization
topic Machine Learning
url https://arxiv.org/abs/2411.09847