Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Improve low-level reconstruction pipeline #17

Open
PaulScotti opened this issue May 31, 2023 · 1 comment
Open

Improve low-level reconstruction pipeline #17

PaulScotti opened this issue May 31, 2023 · 1 comment
Labels
enhancement New feature or request hard-difficulty help wanted Extra attention is needed

Comments

@PaulScotti
Copy link
Collaborator

PaulScotti commented May 31, 2023

Can you figure out a way to beat our current metrics (for low-level pipeline) of .456 (PixCorr) and .493 (SSIM) for subject 1? Use any method you can think of to try to improve upon the current approach.

Maybe mapping to a different embedding space than Stable Diffusion's variational autoencoder? Or adopting a novel training strategy? Could even consider a ControlNet approach with multi-token textual inversion (let me know in advance if you go down that path)

One possibility: Brain-Diffuser has a low-level pipeline that maps to vdvae pretrained on imagenet-64. There is a new vdvae that came out that maps to imagenet-256. Might work better? https://github.com/ericl122333/latent-vae

@PaulScotti PaulScotti added enhancement New feature or request help wanted Extra attention is needed hard-difficulty labels May 31, 2023
@mihirneal
Copy link

Hey Paul, I'll be working on this issue. Currently looking towards using ControlNet in the perceptual pipeline, like the one used in CMVDM. Will keep you updated.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request hard-difficulty help wanted Extra attention is needed
Projects
None yet
Development

No branches or pull requests

2 participants