facebookresearch / facebookresearch/aepsych

Implement baseline linear model

Open
#195 0 comments 0 reactions 0 assignees View on GitHub
Dominant language
HTML
Stars
75
Forks
58
PR merge metrics
No merged PRs in 30d

Description

Many classical psychophysics models are simple linear models or linear models with a polynomial basis. We should include them as baselines in AEPsych. For example, psignifit contains many of them (https://psignifit.sourceforge.net/PSYCHOMETRICFUNCTIONS.html). In the language of the psignifit docs, our "core" is a GP model and our sigmoids are the various objectives here https://github.com/facebookresearch/aepsych/blob/main/aepsych/acquisition/objective.py.

To implement a basic linear model code, we will need a class that extends `AEPsychMixin` that uses a latent linear model with learnable parameters instead of a GP. At minimum, it needs to support a `fit` method, and a `posterior` method so that we can integrate it with acquisition. This issue will require a bit of research to understand where and how to implement everything -- we are happy to support.

Contributor guide

Open the contributing guide

Research direction

Start by reading aepsych/acquisition/objective.py and tracing how the existing GP model integrates with acquisition. Then locate AEPsychMixin and the current model interfaces to determine how a latent linear model should expose fit and posterior. Done means a baseline linear or polynomial-basis model supports both methods and can be integrated with acquisition.

Written by the indexing model from the issue text.

Assessment

Tech stack
python
Domain
machine-learning
Issue type
Feature
Difficulty
5/5
Estimated time
Over a week
Activity status
Stale
Clarity
Mostly clear
Newbie friendliness
35/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.