Parameterize autoencoder architecture
Nobody has claimed this yet.
- 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
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- 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