Giter VIP home page Giter VIP logo

gttn-ngttn's Introduction

Learning Linear and Nonlinear Low-Rank Structure in Multi-Task Learning

Prerequisites

  • Python3
  • PyTorch ==1.6.0
  • Numpy

Datasets

  • fc7 layer feature:
    • office_31_fc7.txt
    • image_clef_fc7.txt
    • office_Home_fc7.txt
    • office-catech_fc7.txt
    • DomainNet_fc7.txt
  • pool5 layer feature:
    • office_31_pool5.txt
    • image_clef_pool5.txt
    • office_Home_pool5.txt
    • office-catech_pool5.txt
    • DomainNet_pool5.txt

Download the pre-processed fc7 layer feature and pool5 layer feature from below link.

https://drive.google.com/drive/folders/1mAc2MPMIzChruQ6SBUC1eHB3XLSaDXvK?usp=sharing

Training

  1. Downloading the dataset(s) from above link.

  2. Run the experiment(s) (task fc7 layer as example):

    cd fc7
    python main.py

    To run this experiment, you need to modify the path datafile = 'xxx.txt' before you run python main.py.

Citation

If you use this code for your research, please consider citing:

@inproceedings{zhang2021multi,
  title={Multi-task learning via generalized tensor trace norm},
  author={Zhang, Yi and Zhang, Yu and Wang, Wei},
  booktitle={Proceedings of the 27th ACM SIGKDD Conference on Knowledge Discovery \& Data Mining},
  pages={2254--2262},
  year={2021}
}

@article{zhang2022learning,
  title={Learning Linear and Nonlinear Low-Rank Structure in Multi-Task Learning},
  author={Zhang, Yi and Zhang, Yu and Wang, Wei},
  journal={IEEE Transactions on Knowledge and Data Engineering},
  year={2022},
  publisher={IEEE}
}

Contact

If you have any problem about our code, feel free to contact [email protected].

gttn-ngttn's People

Contributors

pengxin-guo avatar

Stargazers

 avatar  avatar  avatar

Watchers

 avatar

Forkers

ws-jiang

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    ๐Ÿ–– Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. ๐Ÿ“Š๐Ÿ“ˆ๐ŸŽ‰

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google โค๏ธ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.