FairSTG: Countering performance heterogeneity via collaborative sample-level optimization

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Lin, Gengyu, Zhou, Zhengyang, Huang, Qihe, Yang, Kuo, Cheng, Shifen, Wang, Yang
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866914719678332928
author Lin, Gengyu
Zhou, Zhengyang
Huang, Qihe
Yang, Kuo
Cheng, Shifen
Wang, Yang
author_facet Lin, Gengyu
Zhou, Zhengyang
Huang, Qihe
Yang, Kuo
Cheng, Shifen
Wang, Yang
contents Spatiotemporal learning plays a crucial role in mobile computing techniques to empower smart cites. While existing research has made great efforts to achieve accurate predictions on the overall dataset, they still neglect the significant performance heterogeneity across samples. In this work, we designate the performance heterogeneity as the reason for unfair spatiotemporal learning, which not only degrades the practical functions of models, but also brings serious potential risks to real-world urban applications. To fix this gap, we propose a model-independent Fairness-aware framework for SpatioTemporal Graph learning (FairSTG), which inherits the idea of exploiting advantages of well-learned samples to challenging ones with collaborative mix-up. Specifically, FairSTG consists of a spatiotemporal feature extractor for model initialization, a collaborative representation enhancement for knowledge transfer between well-learned samples and challenging ones, and fairness objectives for immediately suppressing sample-level performance heterogeneity. Experiments on four spatiotemporal datasets demonstrate that our FairSTG significantly improves the fairness quality while maintaining comparable forecasting accuracy. Case studies show FairSTG can counter both spatial and temporal performance heterogeneity by our sample-level retrieval and compensation, and our work can potentially alleviate the risks on spatiotemporal resource allocation for underrepresented urban regions.
format Preprint
id arxiv_https___arxiv_org_abs_2403_12391
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle FairSTG: Countering performance heterogeneity via collaborative sample-level optimization
Lin, Gengyu
Zhou, Zhengyang
Huang, Qihe
Yang, Kuo
Cheng, Shifen
Wang, Yang
Machine Learning
Artificial Intelligence
Spatiotemporal learning plays a crucial role in mobile computing techniques to empower smart cites. While existing research has made great efforts to achieve accurate predictions on the overall dataset, they still neglect the significant performance heterogeneity across samples. In this work, we designate the performance heterogeneity as the reason for unfair spatiotemporal learning, which not only degrades the practical functions of models, but also brings serious potential risks to real-world urban applications. To fix this gap, we propose a model-independent Fairness-aware framework for SpatioTemporal Graph learning (FairSTG), which inherits the idea of exploiting advantages of well-learned samples to challenging ones with collaborative mix-up. Specifically, FairSTG consists of a spatiotemporal feature extractor for model initialization, a collaborative representation enhancement for knowledge transfer between well-learned samples and challenging ones, and fairness objectives for immediately suppressing sample-level performance heterogeneity. Experiments on four spatiotemporal datasets demonstrate that our FairSTG significantly improves the fairness quality while maintaining comparable forecasting accuracy. Case studies show FairSTG can counter both spatial and temporal performance heterogeneity by our sample-level retrieval and compensation, and our work can potentially alleviate the risks on spatiotemporal resource allocation for underrepresented urban regions.
title FairSTG: Countering performance heterogeneity via collaborative sample-level optimization
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2403.12391