{"about":{"site":"https://codewithpapers.app","non_affiliation":"Code with Papers and Syntology are not affiliated with, endorsed by, or sponsored by Papers with Code, Meta, or the pwc-archive mirror.","licence":"CC BY-SA 4.0","licence_url":"https://creativecommons.org/licenses/by-sa/4.0/legalcode","attribution":"https://codewithpapers.app/attribution","modified":"archive material modified by Syntology; see the attribution page"},"url":"/paper/improving-vae-based-molecular-representations","title":"Improving VAE based molecular representations for compound property prediction","arxiv_id":"2201.04929","date":"2022-01-13","proceeding":null,"authors":["A. Tevosyan","L. Khondkaryan","H. Khachatrian","G. Tadevosyan","L. Apresyan","N. Babayan","H. Stopper","Z. Navoyan"],"abstract":"Collecting labeled data for many important tasks in chemoinformatics is time consuming and requires expensive experiments. In recent years, machine learning has been used to learn rich representations of molecules using large scale unlabeled molecular datasets and transfer the knowledge to solve the more challenging tasks with limited datasets. Variational autoencoders are one of the tools that have been proposed to perform the transfer for both chemical property prediction and molecular generation tasks. In this work we propose a simple method to improve chemical property prediction performance of machine learning models by incorporating additional information on correlated molecular descriptors in the representations learned by variational autoencoders. We verify the method on three property prediction asks. We explore the impact of the number of incorporated descriptors, correlation between the descriptors and the target properties, sizes of the datasets etc. Finally, we show the relation between the performance of property prediction models and the distance between property prediction dataset and the larger unlabeled dataset in the representation space.","url_abs":"https://arxiv.org/abs/2201.04929v3","url_pdf":"https://arxiv.org/pdf/2201.04929v3.pdf","source":{"archive":"pwc-archive (Hugging Face), CC BY-SA 4.0","snapshot":"2025-07-28","licence_url":"https://creativecommons.org/licenses/by-sa/4.0/legalcode","row_kind":"abstracts"},"code_links":[{"paper_slug":"improving-vae-based-molecular-representations","repo_url":"https://github.com/znavoyan/vae-embeddings","is_official":1,"mentioned_in_paper":1,"mentioned_in_github":1,"framework":"pytorch","reach":{"status":"ok","spdx":"Apache-2.0"}},{"paper_slug":"improving-vae-based-molecular-representations","repo_url":"https://github.com/znavoyan/vaedatasets","is_official":1,"mentioned_in_paper":1,"mentioned_in_github":0,"framework":"none","reach":{"status":"ok"}}],"tasks":[{"task_slug":"machine-learning","task_name":"BIG-bench Machine Learning"},{"task_slug":"prediction","task_name":"Prediction"},{"task_slug":"property-prediction","task_name":"Property Prediction"}],"methods":[],"datasets_introduced":[],"methods_introduced":[],"results":[],"syntology":{"atlas_url":"https://app.syntology.ai/?focus=2201.04929","mcp":{"get_harvested_code_for_paper":{"arxiv_id":"2201.04929"}},"developers":"https://syntology.ai/developers","read_at":"2026-09-24T18:15:14+00:00","read_at_is":"when the build read Syntology's graph, not when any sample ran","claim":"Per-sample execution status on synthesized fixtures; not a correctness claim about the paper. Samples come from repositories linked to the paper, official or community; repo_kind says which.","repos":[{"provenance":"external:paperswithcode_snapshot_2025-07-28","url":"https://github.com/znavoyan/vaedatasets","reach":{"status":"ok"}},{"provenance":"external:paperswithcode_snapshot_2025-07-28","url":"https://github.com/znavoyan/vae-embeddings","reach":{"status":"ok","spdx":"Apache-2.0"}}],"summary":{"unverified":10},"by_repo_kind":{"official":{"samples":10,"ran":0,"repositories":1}},"repo_kind_vocabulary":{"official":"The archive marks this repository official for the paper","named_in_paper":"The archive records that the paper mentions this repository; it is not marked official","listed":"In the archive's code links for this paper, not marked official and not recorded as mentioned in the paper","found_in_text":"Syntology found this repository in the paper's own text; whether it is the authors' implementation is not asserted","community":"Not in the archive's code links for this paper; a community repository Syntology harvested"},"n_pointer_only_for_licence":0,"samples":[{"code_sha256_prefix":"424b8912cfe83a0e","entry":"adjust_learning_rate","repo":"znavoyan/vae-embeddings","repo_kind":"official","path":"pvae/model_property2.py","file_url":"https://github.com/znavoyan/vae-embeddings/blob/HEAD/pvae/model_property2.py","link_basis":"harvester_set","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"424b8912cfe83a0e"}},{"code_sha256_prefix":"ff750faf19b66abc","entry":"batch2tensor","repo":"znavoyan/vae-embeddings","repo_kind":"official","path":"pvae/model_property2.py","file_url":"https://github.com/znavoyan/vae-embeddings/blob/HEAD/pvae/model_property2.py","link_basis":"harvester_set","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"ff750faf19b66abc"}},{"code_sha256_prefix":"adafd809fa1dd2eb","entry":"kl_loss","repo":"znavoyan/vae-embeddings","repo_kind":"official","path":"chemical_vae/chemvae/train_vae.py","file_url":"https://github.com/znavoyan/vae-embeddings/blob/HEAD/chemical_vae/chemvae/train_vae.py","link_basis":"harvester_set","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"adafd809fa1dd2eb"}},{"code_sha256_prefix":"f4d0cf89b5826b22","entry":"load_encoder","repo":"znavoyan/vae-embeddings","repo_kind":"official","path":"chemical_vae/chemvae/models.py","file_url":"https://github.com/znavoyan/vae-embeddings/blob/HEAD/chemical_vae/chemvae/models.py","link_basis":"harvester_set","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"f4d0cf89b5826b22"}},{"code_sha256_prefix":"477556f5d7718b5c","entry":"load_params","repo":"znavoyan/vae-embeddings","repo_kind":"official","path":"chemical_vae/chemvae/hyperparameters.py","file_url":"https://github.com/znavoyan/vae-embeddings/blob/HEAD/chemical_vae/chemvae/hyperparameters.py","link_basis":"harvester_set","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"477556f5d7718b5c"}},{"code_sha256_prefix":"35a9e12221a43457","entry":"no_schedule","repo":"znavoyan/vae-embeddings","repo_kind":"official","path":"chemical_vae/chemvae/mol_callbacks.py","file_url":"https://github.com/znavoyan/vae-embeddings/blob/HEAD/chemical_vae/chemvae/mol_callbacks.py","link_basis":"harvester_set","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"35a9e12221a43457"}},{"code_sha256_prefix":"9ec4e9feb8c0fd10","entry":"sample","repo":"znavoyan/vae-embeddings","repo_kind":"official","path":"chemical_vae/chemvae/mol_callbacks.py","file_url":"https://github.com/znavoyan/vae-embeddings/blob/HEAD/chemical_vae/chemvae/mol_callbacks.py","link_basis":"harvester_set","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"9ec4e9feb8c0fd10"}},{"code_sha256_prefix":"ae4e7b0062e35c1c","entry":"sampled_rnn","repo":"znavoyan/vae-embeddings","repo_kind":"official","path":"chemical_vae/chemvae/sampled_rnn_tf.py","file_url":"https://github.com/znavoyan/vae-embeddings/blob/HEAD/chemical_vae/chemvae/sampled_rnn_tf.py","link_basis":"harvester_set","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"ae4e7b0062e35c1c"}},{"code_sha256_prefix":"77da6e54bdddeda7","entry":"sigmoid_schedule","repo":"znavoyan/vae-embeddings","repo_kind":"official","path":"chemical_vae/chemvae/mol_callbacks.py","file_url":"https://github.com/znavoyan/vae-embeddings/blob/HEAD/chemical_vae/chemvae/mol_callbacks.py","link_basis":"harvester_set","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"77da6e54bdddeda7"}},{"code_sha256_prefix":"2c9ee1610b7b5f61","entry":"to_var","repo":"znavoyan/vae-embeddings","repo_kind":"official","path":"pvae/model_property2.py","file_url":"https://github.com/znavoyan/vae-embeddings/blob/HEAD/pvae/model_property2.py","link_basis":"harvester_set","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"2c9ee1610b7b5f61"}}]},"arxiv_metadata":null,"syntology_extracted_results":null}