ins-amu / vbjax

A nascent Jax-based package for virtual brain modeling.
Apache License 2.0
7 stars 2 forks source link

Basis embeddings should be sparse #31

Open maedoc opened 1 year ago

maedoc commented 1 year ago

Sparse regression is a clear improvement for interpretability of embedding techniques, so adding the Lasso

image

ridge regression

image

or maybe just the elastic net 🐟

image

(from https://cims.nyu.edu/~cfgranda/pages/OBDA_spring16/material/sparse_regression.pdf) anyway of controlling the number of non zero weights is helpful.

maedoc commented 1 year ago

These penalties could be applied to the MLP flows as well.