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

Implementation of CNN for IceCube #771

Open
sevmag opened this issue Nov 21, 2024 · 1 comment
Open

Implementation of CNN for IceCube #771

sevmag opened this issue Nov 21, 2024 · 1 comment
Labels
feature New feature or request

Comments

@sevmag
Copy link

sevmag commented Nov 21, 2024

Implement the CNN from this paper

Is your feature request related to a problem? Please describe.
Needed for direct comparison between the CNN and other architectures
all within GraphNet trained on the same data.

Describe the solution you'd like
To do this one need to create a new NodeDefinition similar to the one described in #770. Additionally,
it needs a model that will be used as the backbone argument of the StandardModel.
The first thing its forward() function should call is a function that maps the graph resulting from
the new NodeDefinition into the necessary format of the new CNN architecture. After that the forward
should run through the CNN Architecture like expected.

Can someone assign me to this?

@sevmag sevmag added the feature New feature or request label Nov 21, 2024
@pweigel
Copy link
Collaborator

pweigel commented Nov 22, 2024

This architecture is public, you can find it here: https://github.com/icecube/dnn_reco

They use tensorflow and use special hexagonal convolutions which leverage the (approximate) symmetries in IceCube. You can find the code for those convolutions and other utilities here: https://github.com/icecube/TFScripts

Hope this is useful!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
feature New feature or request
Projects
None yet
Development

No branches or pull requests

2 participants