Reverse Engineering DNNs with JIT GEMM Libraries

What is JAXED? JAXED [JIT + AXED] is a security attack targeted reverse engineering DNNs (Deep Neural Network) architectures using JIT optimized GEMM libraries. Hyperparameters specify the model architecture of a DNN. E.g.:- number of layers, number of channels in a filter etc. In our latest work, we exploit a novel side channel exposed during … Continue reading Reverse Engineering DNNs with JIT GEMM Libraries

Runtime Code Generation for Convolutions

marlin-lib

Summary Our work MARLIN (Matrix Multiplication through Reduced Load Instructions), is now available on github. MARLIN is a runtime code generation library for convolution kernels. The paper won the first place in graduate student category in CGO 2021 (International Symposium on Code Generation and Optimization). Here are the links for the 10 minute presentation talk … Continue reading Runtime Code Generation for Convolutions