Skip to content

Implementation of SimCLR model using pytorch for using contrastive learning over shapenet dataset

License

Notifications You must be signed in to change notification settings

mananchawla2005/contrastivelearning

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 

Repository files navigation

Contrastive Learning using Point Cloud

Model

This repository explores the task of contrastive learning applied to 3d point clouds using SimCLR model.

  • SimCLR MODEL:
    simclr-general-architecture
  • NTXENT Loss: image The above image is the mathematical formula for NTXENT Loss introduced by SimCLR model. The numerator comprises of taking consine similarity of the the positive pairs which would lie between -1 to 1. It is then divided by a constant T or temperature and exponential of the value is taken to make it positive. The denominator comprises of summision of exponential of cosine similarity of negative pairs from 0 to 2N ( 2N for augmented pairs ). We dont sum the denominator for k=i ( i.e same as numerator ). Logarithm of whole thing is taken to soften the loss followed by a negative sign.

Results

image image image image

About

Implementation of SimCLR model using pytorch for using contrastive learning over shapenet dataset

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages