PFed-Signal: An ADR Prediction Model based on Federated Learning

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Li, Tao, Li, Peilin, Lu, Kui, Wang, Yilei, Shang, Junliang, Li, Guangshun, Zhou, Huiyu
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866914223444983808
author Li, Tao
Li, Peilin
Lu, Kui
Wang, Yilei
Shang, Junliang
Li, Guangshun
Zhou, Huiyu
author_facet Li, Tao
Li, Peilin
Lu, Kui
Wang, Yilei
Shang, Junliang
Li, Guangshun
Zhou, Huiyu
contents The adverse drug reactions (ADRs) predicted based on the biased records in FAERS (U.S. Food and Drug Administration Adverse Event Reporting System) may mislead diagnosis online. Generally, such problems are solved by optimizing reporting odds ratio (ROR) or proportional reporting ratio (PRR). However, these methods that rely on statistical methods cannot eliminate the biased data, leading to inaccurate signal prediction. In this paper, we propose PFed-signal, a federated learning-based signal prediction model of ADR, which utilizes the Euclidean distance to eliminate the biased data from FAERS, thereby improving the accuracy of ADR prediction. Specifically, we first propose Pfed-Split, a method to split the original dataset into a split dataset based on ADR. Then we propose ADR-signal, an ADR prediction model, including a biased data identification method based on federated learning and an ADR prediction model based on Transformer. The former identifies the biased data according to the Euclidean distance and generates a clean dataset by deleting the biased data. The latter is an ADR prediction model based on Transformer trained on the clean data set. The results show that the ROR and PRR on the clean dataset are better than those of the traditional methods. Furthermore, the accuracy rate, F1 score, recall rate and AUC of PFed-Signal are 0.887, 0.890, 0.913 and 0.957 respectively, which are higher than the baselines.
format Preprint
id arxiv_https___arxiv_org_abs_2512_23262
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle PFed-Signal: An ADR Prediction Model based on Federated Learning
Li, Tao
Li, Peilin
Lu, Kui
Wang, Yilei
Shang, Junliang
Li, Guangshun
Zhou, Huiyu
Machine Learning
The adverse drug reactions (ADRs) predicted based on the biased records in FAERS (U.S. Food and Drug Administration Adverse Event Reporting System) may mislead diagnosis online. Generally, such problems are solved by optimizing reporting odds ratio (ROR) or proportional reporting ratio (PRR). However, these methods that rely on statistical methods cannot eliminate the biased data, leading to inaccurate signal prediction. In this paper, we propose PFed-signal, a federated learning-based signal prediction model of ADR, which utilizes the Euclidean distance to eliminate the biased data from FAERS, thereby improving the accuracy of ADR prediction. Specifically, we first propose Pfed-Split, a method to split the original dataset into a split dataset based on ADR. Then we propose ADR-signal, an ADR prediction model, including a biased data identification method based on federated learning and an ADR prediction model based on Transformer. The former identifies the biased data according to the Euclidean distance and generates a clean dataset by deleting the biased data. The latter is an ADR prediction model based on Transformer trained on the clean data set. The results show that the ROR and PRR on the clean dataset are better than those of the traditional methods. Furthermore, the accuracy rate, F1 score, recall rate and AUC of PFed-Signal are 0.887, 0.890, 0.913 and 0.957 respectively, which are higher than the baselines.
title PFed-Signal: An ADR Prediction Model based on Federated Learning
topic Machine Learning
url https://arxiv.org/abs/2512.23262