Towards Interpretable Automated Machine Learning for STEM Career Prediction

Ruitao Liu, Aixin Tan
<span title="2020-08-22">2020</span> <i title="Zenodo"> Zenodo </i> &nbsp;
In this paper, we describe our solution to predict student STEM career choices during the 2017 ASSISTments Datamining Competition. We built a machine learning system that automatically reformats the data set, generates new features and prunes redundant ones, and performs model and feature selection. We designed the system to automatically find a model that optimizes prediction performance, yet the final model is a simple logistic regression that allows researchers to discover important features
more &raquo; ... and study their effects on STEM career choices. We also compared our method to other methods, which revealed that the key to good prediction is proper feature enrichment in the beginning stage of the data analysis, while feature selection in a later stage allows a simpler final model.
<span class="external-identifiers"> <a target="_blank" rel="external noopener noreferrer" href="https://doi.org/10.5281/zenodo.4008052">doi:10.5281/zenodo.4008052</a> <a target="_blank" rel="external noopener" href="https://fatcat.wiki/release/vwzgnocnhrei7mchsrx64nooku">fatcat:vwzgnocnhrei7mchsrx64nooku</a> </span>
<a target="_blank" rel="noopener" href="https://web.archive.org/web/20200830111316/https://zenodo.org/record/4008052/files/543829050.pdf" title="fulltext PDF download" data-goatcounter-click="serp-fulltext" data-goatcounter-title="serp-fulltext"> <button class="ui simple right pointing dropdown compact black labeled icon button serp-button"> <i class="icon ia-icon"></i> Web Archive [PDF] <div class="menu fulltext-thumbnail"> <img src="https://blobs.fatcat.wiki/thumbnail/pdf/e2/4f/e24f816d056f736563f8b7ed3233940ad801c735.180px.jpg" alt="fulltext thumbnail" loading="lazy"> </div> </button> </a> <a target="_blank" rel="external noopener noreferrer" href="https://doi.org/10.5281/zenodo.4008052"> <button class="ui left aligned compact blue labeled icon button serp-button"> <i class="unlock alternate icon" style="background-color: #fb971f;"></i> zenodo.org </button> </a>