ML4GW / ML4GW/DeepClean

Parameterize autoencoder architecture

Open
#34 1 comment 0 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

hyperparameters research topic
Dominant language
Python
Stars
10
Forks
6
PR merge metrics
No merged PRs in 30d

Description

The current implementation of the autoencoder architecture make picking the depth dynamically difficult because the paddings in the inverse transposes and the padding/kernel size in the last convolution layer need to be picked just-so in order that the output length magically matches the input length. This makes exploring depth or parameter count automatically as a hyperparameter more difficult.

One solution to this might be to do dynamic shape inference on the output of the transpose convolutions, from which point it should be (I think) reasonably straightforward algebra to infer the length of kernel required in the last convolution to make the lengths work out. Keras would have this functionality, but unfortunately with Torch I think we'll just have to work out the math ourselves (though I'd LOVE to be proven wrong here).

Contributor guide

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Research direction

Start from the current autoencoder implementation and trace the transpose-convolution output shapes in Torch. Work out how architecture depth and the final convolution parameters can be selected so the output length matches the input length; done means depth and parameter count can be explored automatically without hand-tuned padding.

Written by the indexing model from the issue text.

Assessment

Tech stack
python, pytorch
Domain
machine-learning
Issue type
Feature
Difficulty
5/5
Estimated time
Over a week
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
25/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.