/PLR_Mindspore_zj

Primary LanguagePythonApache License 2.0Apache-2.0

PLR(AAAI2023)

This is the MindSpore implementation of PLR in the following paper.

AAAI 2023: Partial-Label Regression

This paper provides the first attempt to investigate partial-label regression, where each training example is annotated with a set of real-valued candidate labels.

  1. Propose a simple baseline method that takes the average loss incurred by candidate labels as the predictive loss to be minimized for model training.

  2. Propose an identification method that takes the least loss incurred by candidate labels as the predictive loss to be minimized for model training.

  3. Propose a progressive identification method that differentiates candidate labels by associating their incurred losses with progressively updated weights.

  4. Show that the identification method and the progressive identification method are model-consistent, which indicates that the learned model converges to the optimal model.

Our experiments are conducted on seven widely used benchmark regression datasets to test the performance of our PLR, which are Abalone, Airfoil, Auto-mpg, Housing, Concrete, Power-plant and Cpu-act.

Framework

For more information, please check the resources below:

After installing MindSpore via the official website, you can start training and evaluation as follows:

python demo_ms.py